Spaces:
Sleeping
Sleeping
Update src/ai_processor.py
Browse files- src/ai_processor.py +1 -1
src/ai_processor.py
CHANGED
|
@@ -149,7 +149,7 @@ def _medgemma_generate_gpu(prompt: str, model_id: str, max_new_tokens: int, toke
|
|
| 149 |
|
| 150 |
pipe = pipeline(
|
| 151 |
"image-text-to-text",
|
| 152 |
-
model="
|
| 153 |
torch_dtype=torch.bfloat16,
|
| 154 |
device="cuda",
|
| 155 |
)
|
|
|
|
| 149 |
|
| 150 |
pipe = pipeline(
|
| 151 |
"image-text-to-text",
|
| 152 |
+
model="unsloth/medgemma-4b-it-unsloth-bnb-4bit",
|
| 153 |
torch_dtype=torch.bfloat16,
|
| 154 |
device="cuda",
|
| 155 |
)
|