Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -195,10 +195,10 @@ def load_ckpt(input_style):
|
|
195 |
global MultiResNetModel
|
196 |
if input_style == "Sketch" or input_style == "Sketch_Shading":
|
197 |
if input_style == "Sketch":
|
198 |
-
ckpt_path = model_global_path + '/
|
199 |
rank = 128
|
200 |
else:
|
201 |
-
ckpt_path = model_global_path + '/
|
202 |
rank = 128
|
203 |
pretrained_model_name_or_path = 'PixArt-alpha/PixArt-XL-2-1024-MS'
|
204 |
transformer = PixArtTransformer2DModel.from_pretrained(
|
|
|
195 |
global MultiResNetModel
|
196 |
if input_style == "Sketch" or input_style == "Sketch_Shading":
|
197 |
if input_style == "Sketch":
|
198 |
+
ckpt_path = model_global_path + '/sketch/'
|
199 |
rank = 128
|
200 |
else:
|
201 |
+
ckpt_path = model_global_path + '/shading/'
|
202 |
rank = 128
|
203 |
pretrained_model_name_or_path = 'PixArt-alpha/PixArt-XL-2-1024-MS'
|
204 |
transformer = PixArtTransformer2DModel.from_pretrained(
|