Spaces:
Running
on
Zero
Running
on
Zero
support multi lora
Browse files
app.py
CHANGED
@@ -85,7 +85,7 @@ def upload_image_to_r2(image, account_id, access_key, secret_key, bucket_name):
|
|
85 |
return image_file
|
86 |
|
87 |
|
88 |
-
|
89 |
def generate_image(prompt, steps, seed, cfg_scale, width, height, progress):
|
90 |
pipe.to("cuda")
|
91 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
@@ -104,7 +104,7 @@ def generate_image(prompt, steps, seed, cfg_scale, width, height, progress):
|
|
104 |
progress(99, "Generate success!")
|
105 |
return generate_image
|
106 |
|
107 |
-
|
108 |
def run_lora(prompt, cfg_scale, steps, lora_strings, randomize_seed, seed, width, height, lora_scale, upload_to_r2, account_id, access_key, secret_key, bucket, progress=gr.Progress(track_tqdm=True)):
|
109 |
|
110 |
|
@@ -167,7 +167,7 @@ with gr.Blocks(css=css) as demo:
|
|
167 |
with gr.Row():
|
168 |
seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
|
169 |
randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
|
170 |
-
lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=
|
171 |
|
172 |
with gr.Row():
|
173 |
width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
|
|
|
85 |
return image_file
|
86 |
|
87 |
|
88 |
+
|
89 |
def generate_image(prompt, steps, seed, cfg_scale, width, height, progress):
|
90 |
pipe.to("cuda")
|
91 |
generator = torch.Generator(device="cuda").manual_seed(seed)
|
|
|
104 |
progress(99, "Generate success!")
|
105 |
return generate_image
|
106 |
|
107 |
+
@spaces.GPU
|
108 |
def run_lora(prompt, cfg_scale, steps, lora_strings, randomize_seed, seed, width, height, lora_scale, upload_to_r2, account_id, access_key, secret_key, bucket, progress=gr.Progress(track_tqdm=True)):
|
109 |
|
110 |
|
|
|
167 |
with gr.Row():
|
168 |
seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
|
169 |
randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
|
170 |
+
lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=1, step=0.01, value=0.5)
|
171 |
|
172 |
with gr.Row():
|
173 |
width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
|