Hugging Face's logo

Spaces:
rynmurdock
/
generative_recsys
Sleeping

App Files Community
5
rynmurdock commited on
Commit
d2048d8
·
1 Parent(s): a586179

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -130,7 +130,7 @@ pipe.to(device=DEVICE)
130
  from transformers import AutoProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig
131
 
132
  quantization_config = BitsAndBytesConfig(load_in_4bit=True)
133
- pali = PaliGemmaForConditionalGeneration.from_pretrained('google/paligemma-3b-pt-224', torch_dtype=dtype, quantization_config=quantization_config).eval()
134
  processor = AutoProcessor.from_pretrained('google/paligemma-3b-pt-224')
135
 
136
 
 
130
  from transformers import AutoProcessor, PaliGemmaForConditionalGeneration, BitsAndBytesConfig
131
 
132
  quantization_config = BitsAndBytesConfig(load_in_4bit=True)
133
+ pali = PaliGemmaForConditionalGeneration.from_pretrained('google/paligemma-3b-pt-224', torch_dtype=dtype,).eval().to('cuda')
134
  processor = AutoProcessor.from_pretrained('google/paligemma-3b-pt-224')
135
 
136