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 = '
|
| 199 |
rank = 128
|
| 200 |
else:
|
| 201 |
-
ckpt_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 + '/ckpt/sketch/'
|
| 199 |
rank = 128
|
| 200 |
else:
|
| 201 |
+
ckpt_path = model_global_path + '/ckpt/shading/'
|
| 202 |
rank = 128
|
| 203 |
pretrained_model_name_or_path = 'PixArt-alpha/PixArt-XL-2-1024-MS'
|
| 204 |
transformer = PixArtTransformer2DModel.from_pretrained(
|