JunhaoZhuang commited on
Commit
00b597f
·
verified ·
1 Parent(s): 8804bb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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 + '/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(
 
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(