prithivMLmods commited on
Commit
cd0adff
·
verified ·
1 Parent(s): 2d99b82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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-4b-it" # alternative: google/gemma-3-12b-it
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()