AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

run make style

Browse Source
This commit is contained in:
Pham Hong Vinh
2024-01-12 12:33:53 +07:00
parent 4191fc30f4
commit 362461ba04
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
examples/dreambooth/train_dreambooth.py
View File

@@ -996,14 +996,11 @@ def main(args):
)
if unwrap_model(unet).dtype != torch.float32:
raise ValueError(
f"Unet loaded as datatype {unwrap_model(unet).dtype}. {low_precision_error_string}"
)
raise ValueError(f"Unet loaded as datatype {unwrap_model(unet).dtype}. {low_precision_error_string}")
if args.train_text_encoder and unwrap_model(text_encoder).dtype != torch.float32:
raise ValueError(
f"Text encoder loaded as datatype {unwrap_model(text_encoder).dtype}."
f" {low_precision_error_string}"
f"Text encoder loaded as datatype {unwrap_model(text_encoder).dtype}." f" {low_precision_error_string}"
)
# Enable TF32 for faster training on Ampere GPUs,