Update vision.py
This commit is contained in:
parent
a3d6b3a4bf
commit
067db89dc3
1 changed files with 2 additions and 2 deletions
|
|
@ -521,8 +521,8 @@ class FastBaseModel:
|
|||
)
|
||||
else:
|
||||
print(
|
||||
f"Unsloth: Using full float32 full finetuning.\n"
|
||||
f"To enable bfloat16 training to reduce VRAM usage by 50% albeit with a slightly higher loss, "\
|
||||
f"Unsloth: Using full float32 full finetuning. "
|
||||
f"To enable bfloat16 training to reduce VRAM usage by 50% albeit with a slightly higher loss, do:\n"\
|
||||
"use `float32_mixed_precision = False` during FastLanguageModel.from_pretrained"
|
||||
)
|
||||
os.environ["UNSLOTH_BFLOAT16_MIXED_PRECISION"] = "1"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue