Spaces:
Runtime error
Runtime error
dafajudin
commited on
Commit
•
996ed80
1
Parent(s):
adffb04
update code
Browse files- app.py +5 -5
- index.html +1 -1
app.py
CHANGED
@@ -35,11 +35,11 @@ if USE_QLORA or USE_LORA:
|
|
35 |
)
|
36 |
|
37 |
# Model Idefics2
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
|
44 |
processor = AutoProcessor.from_pretrained(
|
45 |
"HuggingFaceM4/idefics2-8b",
|
|
|
35 |
)
|
36 |
|
37 |
# Model Idefics2
|
38 |
+
model = Idefics2ForConditionalGeneration.from_pretrained(
|
39 |
+
"jihadzakki/idefics2-8b-vqarad-delta",
|
40 |
+
torch_dtype=torch.float16,
|
41 |
+
quantization_config=bnb_config
|
42 |
+
)
|
43 |
|
44 |
processor = AutoProcessor.from_pretrained(
|
45 |
"HuggingFaceM4/idefics2-8b",
|
index.html
CHANGED
@@ -120,7 +120,7 @@
|
|
120 |
<h3 class="overview-heading"><span class="vl">Model Architecture</span></h3>
|
121 |
<div>
|
122 |
<p class="overview-content">The model is trained using Idefics2-8b.</p>
|
123 |
-
<img class="content-image" src="img/idefics2_architecture.png" alt="model-architecture" />
|
124 |
</div>
|
125 |
</div>
|
126 |
</section>
|
|
|
120 |
<h3 class="overview-heading"><span class="vl">Model Architecture</span></h3>
|
121 |
<div>
|
122 |
<p class="overview-content">The model is trained using Idefics2-8b.</p>
|
123 |
+
<img class="content-image" src="https://raw.githubusercontent.com/Kalbe-x-Bangkit/C24-RM-Kalbe-Bangkit/main/img/idefics2_architecture.png" alt="model-architecture" />
|
124 |
</div>
|
125 |
</div>
|
126 |
</section>
|