Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -60,13 +60,13 @@ def scrape_lora_link(url):
|
|
60 |
|
61 |
def enable_lora(lora_scale,lora_add):
|
62 |
if not lora_add:
|
63 |
-
gr.Info("No Lora Loaded, Use basemodel"
|
64 |
else:
|
65 |
url = f'https://huggingface.co/{lora_add}/tree/main'
|
66 |
lora_name = scrape_lora_link(url)
|
67 |
pipe.load_lora_weights(lora_add, weight_name=lora_name)
|
68 |
pipe.fuse_lora(lora_scale=lora_scale)
|
69 |
-
gr.Info(f"{lora_add} Loaded"
|
70 |
|
71 |
@spaces.GPU()
|
72 |
def generate_image(
|
|
|
60 |
|
61 |
def enable_lora(lora_scale,lora_add):
|
62 |
if not lora_add:
|
63 |
+
gr.Info("No Lora Loaded, Use basemodel")
|
64 |
else:
|
65 |
url = f'https://huggingface.co/{lora_add}/tree/main'
|
66 |
lora_name = scrape_lora_link(url)
|
67 |
pipe.load_lora_weights(lora_add, weight_name=lora_name)
|
68 |
pipe.fuse_lora(lora_scale=lora_scale)
|
69 |
+
gr.Info(f"{lora_add} Loaded")
|
70 |
|
71 |
@spaces.GPU()
|
72 |
def generate_image(
|