Commit ·
82f82ac
1
Parent(s): 58a4476
Use dtype instead of deprecated torch_dtype in model_kwargs
Browse filesCo-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
- 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 |
-
|
| 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 |
|