I need to rename tis variable to use the default arg...

d8ahazard · Nov 9, 2022 · ef4c80a · ef4c80a
1 parent e2bb431
commit ef4c80a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dreambooth/train_dreambooth.py b/dreambooth/train_dreambooth.py
@@ -731,7 +731,7 @@ def save_weights(step, save_model, save_img):
             if args.train_text_encoder:
                 text_enc_model = accelerator.unwrap_model(text_encoder)
             else:
-                text_enc_model = CLIPTextModel.from_pretrained(args.pretrained_model_name_or_path, subfolder="text_encoder")
+                text_enc_model = CLIPTextModel.from_pretrained(args.working_dir, subfolder="text_encoder")
             scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
             pipeline = StableDiffusionPipeline.from_pretrained(
                 args.working_dir,