false mixed precision

tengshaofeng · Feb 13, 2023 · abbde84 · abbde84
1 parent 9b6df8f
commit abbde84
Showing 1 changed file with 1 addition and 1 deletion.
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,
         )