Spaces:
Build error
Build error
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
|