Update app.py
Browse files
app.py
CHANGED
@@ -66,10 +66,9 @@ def enable_lora(lora_scale, lora_in, lora_add):
|
|
66 |
lora_add = lora_in
|
67 |
url = f'https://huggingface.co/{lora_add}/tree/main'
|
68 |
lora_name = scrape_lora_link(url)
|
69 |
-
pipe.
|
70 |
pipe.fuse_lora(lora_scale=lora_scale)
|
71 |
|
72 |
-
@spaces.GPU()
|
73 |
def generate_image(
|
74 |
prompt:str,
|
75 |
width:int=768,
|
@@ -104,7 +103,7 @@ def generate_image(
|
|
104 |
|
105 |
return image, seed
|
106 |
|
107 |
-
|
108 |
def gen(
|
109 |
prompt:str,
|
110 |
width:int=768,
|
@@ -210,10 +209,10 @@ with gr.Blocks(css=CSS, js=JS, theme="Nymbo/Nymbo_Theme") as demo:
|
|
210 |
value=1.0,
|
211 |
)
|
212 |
lora_in = gr.Dropdown(
|
213 |
-
choices=["Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration", "Shakker-Labs/AWPortrait-FL"
|
214 |
label="LoRA Model",
|
215 |
info="Load the LoRA",
|
216 |
-
value="
|
217 |
)
|
218 |
lora_add = gr.Textbox(
|
219 |
label="Add Flux LoRA",
|
|
|
66 |
lora_add = lora_in
|
67 |
url = f'https://huggingface.co/{lora_add}/tree/main'
|
68 |
lora_name = scrape_lora_link(url)
|
69 |
+
pipe.load_lora_weights(lora_add, weight_name=lora_name)
|
70 |
pipe.fuse_lora(lora_scale=lora_scale)
|
71 |
|
|
|
72 |
def generate_image(
|
73 |
prompt:str,
|
74 |
width:int=768,
|
|
|
103 |
|
104 |
return image, seed
|
105 |
|
106 |
+
@spaces.GPU()
|
107 |
def gen(
|
108 |
prompt:str,
|
109 |
width:int=768,
|
|
|
209 |
value=1.0,
|
210 |
)
|
211 |
lora_in = gr.Dropdown(
|
212 |
+
choices=["Shakker-Labs/FLUX.1-dev-LoRA-blended-realistic-illustration", "Shakker-Labs/AWPortrait-FL"],
|
213 |
label="LoRA Model",
|
214 |
info="Load the LoRA",
|
215 |
+
value="",
|
216 |
)
|
217 |
lora_add = gr.Textbox(
|
218 |
label="Add Flux LoRA",
|