Spaces:

Didier
/

Vision_language_Gemma3

Running on Zero

Didier commited on Mar 16

Commit

e40f135

verified ·

1 Parent(s): 37f622b

Update vlm.py

Files changed (1) hide show

vlm.py CHANGED Viewed

@@ -14,7 +14,7 @@ import torch
 #
 # Load the model: google/gemma-3-4b-it
 #
-device = 'mps'
 model_id = "google/gemma-3-4b-it"
 processor = AutoProcessor.from_pretrained(model_id, use_fast=True, padding_side="left")
 model = Gemma3ForConditionalGeneration.from_pretrained(

 #
 # Load the model: google/gemma-3-4b-it
 #
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
 model_id = "google/gemma-3-4b-it"
 processor = AutoProcessor.from_pretrained(model_id, use_fast=True, padding_side="left")
 model = Gemma3ForConditionalGeneration.from_pretrained(