Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -104,7 +104,7 @@ dtype = torch.float16 if device.type == "cuda" else torch.float32
|
|
104 |
|
105 |
# GEMMA3-4B MULTIMODAL MODEL
|
106 |
|
107 |
-
gemma3_model_id = "google/gemma-3-
|
108 |
gemma3_model = Gemma3ForConditionalGeneration.from_pretrained(
|
109 |
gemma3_model_id, device_map="auto"
|
110 |
).eval()
|
|
|
104 |
|
105 |
# GEMMA3-4B MULTIMODAL MODEL
|
106 |
|
107 |
+
gemma3_model_id = "google/gemma-3-12b-it" # alternative: google/gemma-3-4b-it
|
108 |
gemma3_model = Gemma3ForConditionalGeneration.from_pretrained(
|
109 |
gemma3_model_id, device_map="auto"
|
110 |
).eval()
|