NYUAD-ComNets commited on
Commit
b84873d
·
verified ·
1 Parent(s): 371a337

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +20 -1
README.md CHANGED
@@ -77,7 +77,7 @@ torch._dynamo.config.disable = True
77
  torch._dynamo.reset()
78
 
79
  # 2. Load your fine-tuned model and processor
80
- model_path = "NYUAD-ComNets/Gemma3_meme"
81
  model, processor = FastVisionModel.from_pretrained(
82
  model_path,
83
  load_in_4bit = True,
@@ -150,6 +150,25 @@ np.save("test_embeddings.npy", embedding_matrix)
150
  # Finetuned Gemma 3 for Inference
151
 
152
  ``` python
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
153
  dataset = load_dataset("QCRI/ArGuard-Task1", split="dev_test")
154
 
155
  ids = []
 
77
  torch._dynamo.reset()
78
 
79
  # 2. Load your fine-tuned model and processor
80
+ model_path = "NYUAD-ComNets/Gemma3_meme_classification"
81
  model, processor = FastVisionModel.from_pretrained(
82
  model_path,
83
  load_in_4bit = True,
 
150
  # Finetuned Gemma 3 for Inference
151
 
152
  ``` python
153
+
154
+ import numpy as np
155
+ import torch
156
+ import torch._dynamo
157
+ from tqdm import tqdm
158
+ from unsloth import FastVisionModel
159
+
160
+ # 1. Disable compiler optimization conflicts
161
+ torch._dynamo.config.disable = True
162
+ torch._dynamo.reset()
163
+
164
+ # 2. Load your fine-tuned model and processor
165
+ model_path = "NYUAD-ComNets/Gemma3_meme_classification"
166
+ model, processor = FastVisionModel.from_pretrained(
167
+ model_path,
168
+ load_in_4bit = True,
169
+ )
170
+ FastVisionModel.for_inference(model)
171
+
172
  dataset = load_dataset("QCRI/ArGuard-Task1", split="dev_test")
173
 
174
  ids = []