Spaces:
Build error
Build error
JiayiGuo821
commited on
Commit
•
cdbeb67
1
Parent(s):
302fb80
Add files
Browse files
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', '
|
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=
|
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
|