run make style

rootonchair · rootonchair · commit 362461ba04d6 · 2024-01-12T12:33:53.000+07:00
diff --git a/examples/dreambooth/train_dreambooth.py b/examples/dreambooth/train_dreambooth.py
@@ -996,14 +996,11 @@ def load_model_hook(models, input_dir):
     )
 
     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,

Original file line number	Diff line number	Diff line change
`@@ -996,14 +996,11 @@ def load_model_hook(models, input_dir):`
`996`	`996`	`)`
`997`	`997`
`998`	`998`	`if unwrap_model(unet).dtype != torch.float32:`
`999`		`- raise ValueError(`
`1000`		`- f"Unet loaded as datatype {unwrap_model(unet).dtype}. {low_precision_error_string}"`
`1001`		`- )`
	`999`	`+ raise ValueError(f"Unet loaded as datatype {unwrap_model(unet).dtype}. {low_precision_error_string}")`
`1002`	`1000`
`1003`	`1001`	`if args.train_text_encoder and unwrap_model(text_encoder).dtype != torch.float32:`
`1004`	`1002`	`raise ValueError(`
`1005`		`- f"Text encoder loaded as datatype {unwrap_model(text_encoder).dtype}."`
`1006`		`- f" {low_precision_error_string}"`
	`1003`	`+ f"Text encoder loaded as datatype {unwrap_model(text_encoder).dtype}." f" {low_precision_error_string}"`
`1007`	`1004`	`)`
`1008`	`1005`
`1009`	`1006`	`# Enable TF32 for faster training on Ampere GPUs,`