false mixed precision

cloneofsimo · cloneofsimo · commit abbde8456ca3 · 2023-02-13T09:04:20.000Z
diff --git a/lora_diffusion/cli_lora_pti.py b/lora_diffusion/cli_lora_pti.py
@@ -924,7 +924,7 @@ def train(
             wandb_log_prompt_cnt=wandb_log_prompt_cnt,
             class_token=class_token,
             train_inpainting=train_inpainting,
-            mixed_precision=True,
+            mixed_precision=False,
             tokenizer=tokenizer,
             clip_ti_decay=clip_ti_decay,
         )

Original file line number	Diff line number	Diff line change
`@@ -924,7 +924,7 @@ def train(`
`924`	`924`	`wandb_log_prompt_cnt=wandb_log_prompt_cnt,`
`925`	`925`	`class_token=class_token,`
`926`	`926`	`train_inpainting=train_inpainting,`
`927`		`- mixed_precision=True,`
	`927`	`+ mixed_precision=False,`
`928`	`928`	`tokenizer=tokenizer,`
`929`	`929`	`clip_ti_decay=clip_ti_decay,`
`930`	`930`	`)`