Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -92,7 +92,7 @@ dtype = torch.float16 if device.type == "cuda" else torch.float32
|
|
92 |
|
93 |
# Gemma3 Model (default for text, image, & video inference)
|
94 |
|
95 |
-
gemma3_model_id = "google/gemma-3-4b-it"
|
96 |
gemma3_model = Gemma3ForConditionalGeneration.from_pretrained(
|
97 |
gemma3_model_id, device_map="auto"
|
98 |
).eval()
|
|
|
92 |
|
93 |
# Gemma3 Model (default for text, image, & video inference)
|
94 |
|
95 |
+
gemma3_model_id = "google/gemma-3-4b-it"
|
96 |
gemma3_model = Gemma3ForConditionalGeneration.from_pretrained(
|
97 |
gemma3_model_id, device_map="auto"
|
98 |
).eval()
|