Spaces:

ktrndy
/

diffusion-image-gen

Running

ktrndy commited on Feb 9

Commit

f308f4a

verified ·

1 Parent(s): 5ac29d1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,7 @@ def infer(
 ):
     generator = torch.Generator(device).manual_seed(seed)
-    ckpt_dir='./output'
     unet_sub_dir = os.path.join(ckpt_dir, "unet")
     text_encoder_sub_dir = os.path.join(ckpt_dir, "text_encoder")

 ):
     generator = torch.Generator(device).manual_seed(seed)
+    ckpt_dir='./model_output'
     unet_sub_dir = os.path.join(ckpt_dir, "unet")
     text_encoder_sub_dir = os.path.join(ckpt_dir, "text_encoder")