Update vision.py
This commit is contained in:
parent
e88cb620ab
commit
6ef73397f2
1 changed files with 1 additions and 0 deletions
|
|
@ -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:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue