JiayiGuo821 commited on
Commit
cdbeb67
1 Parent(s): 302fb80
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -44,7 +44,7 @@ choices.diffuser = OrderedDict([
44
 
45
  choices.lora = OrderedDict([
46
  ['empty', ""],
47
- ['Smooth-LoRA-v1', hf_hub_download('shi-labs/smooth-diffusion-lora', 'pytorch_model.bin')],
48
  ])
49
 
50
  choices.scheduler = OrderedDict([
@@ -286,7 +286,7 @@ class wrapper(object):
286
  self.net.safety_checker = None
287
  if tag_lora != 'empty':
288
  self.net.unet.load_attn_procs(
289
- choices.lora[tag_lora], use_safetensors=False,)
290
  self.tag_diffuser = tag_diffuser
291
  self.tag_lora = tag_lora
292
  return tag_diffuser, tag_lora
 
44
 
45
  choices.lora = OrderedDict([
46
  ['empty', ""],
47
+ ['Smooth-LoRA-v1', hf_hub_download('shi-labs/smooth-diffusion-lora', 'smooth_lora.safetensors')],
48
  ])
49
 
50
  choices.scheduler = OrderedDict([
 
286
  self.net.safety_checker = None
287
  if tag_lora != 'empty':
288
  self.net.unet.load_attn_procs(
289
+ choices.lora[tag_lora], use_safetensors=True,)
290
  self.tag_diffuser = tag_diffuser
291
  self.tag_lora = tag_lora
292
  return tag_diffuser, tag_lora