dafajudin commited on
Commit
996ed80
1 Parent(s): adffb04

update code

Browse files
Files changed (2) hide show
  1. app.py +5 -5
  2. index.html +1 -1
app.py CHANGED
@@ -35,11 +35,11 @@ if USE_QLORA or USE_LORA:
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",
 
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>