cgoodmaker Claude Opus 4.6 commited on
Commit
82f82ac
·
1 Parent(s): 58a4476

Use dtype instead of deprecated torch_dtype in model_kwargs

Browse files

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Files changed (1) hide show
  1. models/medgemma_agent.py +1 -1
models/medgemma_agent.py CHANGED
@@ -229,7 +229,7 @@ class MedGemmaAgent:
229
  self._print("Using CPU")
230
 
231
  model_kwargs = dict(
232
- torch_dtype=torch.bfloat16 if device != "cpu" else torch.float32,
233
  device_map="auto",
234
  )
235
 
 
229
  self._print("Using CPU")
230
 
231
  model_kwargs = dict(
232
+ dtype=torch.bfloat16 if device != "cpu" else torch.float32,
233
  device_map="auto",
234
  )
235