Spaces:

JunhaoZhuang
/

Cobra

Runtime error

JunhaoZhuang commited on Apr 15

Commit

c701af0

verified ·

1 Parent(s): 1468e82

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -201,7 +201,7 @@ def load_ckpt():
             ])
     # seed = 43
     lora_rank = 128
-    pretrained_model_name_or_path = '/mnt/workspace/zhuangjunhao/ColorFlow/PixArt-XL-2-1024-MS/models--PixArt-alpha--PixArt-XL-2-1024-MS/snapshots/b89adadeccd9ead2adcb9fa2825d3fabec48d404'
     transformer = PixArtTransformer2DModel.from_pretrained(
             pretrained_model_name_or_path, subfolder="transformer", revision=None, variant=None

             ])
     # seed = 43
     lora_rank = 128
+    pretrained_model_name_or_path = "PixArt-alpha/PixArt-XL-2-1024-MS"
     transformer = PixArtTransformer2DModel.from_pretrained(
             pretrained_model_name_or_path, subfolder="transformer", revision=None, variant=None