SmartHeal commited on
Commit
f85c4fc
·
verified ·
1 Parent(s): 081b143

Update src/ai_processor.py

Browse files
Files changed (1) hide show
  1. 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="google/medgemma-4b-it",
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
  )