Spaces:
Runtime error
Runtime error
removing bnb for now
Browse files
app.py
CHANGED
|
@@ -123,10 +123,9 @@ pipe.to(device=DEVICE)
|
|
| 123 |
|
| 124 |
#############################################################
|
| 125 |
|
| 126 |
-
from transformers import AutoProcessor, PaliGemmaForConditionalGeneration
|
| 127 |
|
| 128 |
-
|
| 129 |
-
pali = PaliGemmaForConditionalGeneration.from_pretrained('google/paligemma-3b-pt-224', torch_dtype=dtype, quantization_config=quantization_config).eval()
|
| 130 |
processor = AutoProcessor.from_pretrained('google/paligemma-3b-pt-224')
|
| 131 |
|
| 132 |
#pali = torch.compile(pali)
|
|
|
|
| 123 |
|
| 124 |
#############################################################
|
| 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)
|