Hugging Face's logo

Spaces:
rynmurdock
/
generative_recsys
Sleeping

App Files Community
5
rynmurdock commited on
Commit
4769988
•
1 Parent(s): 3802eeb

lol my b

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -125,7 +125,7 @@ pipe.to(device=DEVICE)
125
 
126
  from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
127
 
128
- pali = PaliGemmaForConditionalGeneration.from_pretrained('google/paligemma-3b-pt-224', torch_dtype=dtype).eval()
129
  processor = AutoProcessor.from_pretrained('google/paligemma-3b-pt-224')
130
 
131
  #pali = torch.compile(pali)
 
125
 
126
  from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
127
 
128
+ pali = PaliGemmaForConditionalGeneration.from_pretrained('google/paligemma-3b-pt-224', torch_dtype=dtype).eval().to('cuda')
129
  processor = AutoProcessor.from_pretrained('google/paligemma-3b-pt-224')
130
 
131
  #pali = torch.compile(pali)