Spaces:

timbrooks
/

instruct-pix2pix

Running on T4

holynski commited on Jan 19, 2023

Commit

3c6e4f4

•

1 Parent(s): 2610499

Update edit_app.py

Files changed (1) hide show

edit_app.py CHANGED Viewed

@@ -101,7 +101,7 @@ def main():
     parser = ArgumentParser()
     parser.add_argument("--resolution", default=512, type=int)
     parser.add_argument("--config", default="configs/generate.yaml", type=str)
-    parser.add_argument("--ckpt", default="checkpoints/instruct-pix2pix-00-22000.ckpt", type=str)
     parser.add_argument("--vae-ckpt", default=None, type=str)
     args = parser.parse_args()

     parser = ArgumentParser()
     parser.add_argument("--resolution", default=512, type=int)
     parser.add_argument("--config", default="configs/generate.yaml", type=str)
+    parser.add_argument("--ckpt", default="diffusers/pix2pix-sd/instruct-pix2pix-00-22000.ckpt", type=str)
     parser.add_argument("--vae-ckpt", default=None, type=str)
     args = parser.parse_args()