Update vision.py

This commit is contained in:
Daniel Han 2025-10-30 05:38:16 -07:00
commit 6ef73397f2

View file

@ -948,6 +948,7 @@ class FastBaseModel:
if type(float32_mixed_precision) is bool:
# Respect whatever it was set before
pass
else:
float32_mixed_precision = True
if _get_dtype(dtype_from_config(model.config)) == torch.bfloat16 and full_finetuning: