Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ pipe = diffusers.ZImagePipeline.from_pretrained("dimitribarbot/Z-Image-Turbo-BF1
|
|
| 50 |
|
| 51 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 52 |
|
| 53 |
-
pipe.enable_model_cpu_offload()
|
| 54 |
|
| 55 |
try: # A temp hack for some version diffusers lora loading problem
|
| 56 |
from diffusers.utils.peft_utils import _derive_exclude_modules
|
|
@@ -207,7 +207,7 @@ with gr.Blocks(css=css) as app:
|
|
| 207 |
with gr.Column():
|
| 208 |
with gr.Row():
|
| 209 |
cfg_scale = gr.Slider(label="CFG Scale", minimum=0, maximum=20, step=.1, value=1.0)
|
| 210 |
-
steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=
|
| 211 |
|
| 212 |
with gr.Row():
|
| 213 |
width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
|
|
@@ -216,7 +216,7 @@ with gr.Blocks(css=css) as app:
|
|
| 216 |
with gr.Row():
|
| 217 |
randomize_seed = gr.Checkbox(True, label="Randomize seed")
|
| 218 |
seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
|
| 219 |
-
lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=2.5, step=0.01, value=
|
| 220 |
|
| 221 |
gallery.select(
|
| 222 |
update_selection,
|
|
|
|
| 50 |
|
| 51 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 52 |
|
| 53 |
+
#pipe.enable_model_cpu_offload()
|
| 54 |
|
| 55 |
try: # A temp hack for some version diffusers lora loading problem
|
| 56 |
from diffusers.utils.peft_utils import _derive_exclude_modules
|
|
|
|
| 207 |
with gr.Column():
|
| 208 |
with gr.Row():
|
| 209 |
cfg_scale = gr.Slider(label="CFG Scale", minimum=0, maximum=20, step=.1, value=1.0)
|
| 210 |
+
steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=10)
|
| 211 |
|
| 212 |
with gr.Row():
|
| 213 |
width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
|
|
|
|
| 216 |
with gr.Row():
|
| 217 |
randomize_seed = gr.Checkbox(True, label="Randomize seed")
|
| 218 |
seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=0, randomize=True)
|
| 219 |
+
lora_scale = gr.Slider(label="LoRA Scale", minimum=0, maximum=2.5, step=0.01, value=0.8)
|
| 220 |
|
| 221 |
gallery.select(
|
| 222 |
update_selection,
|