Spaces:
Runtime error
Runtime error
ameerazam08
commited on
Commit
β’
24ce25c
1
Parent(s):
927b744
negative prompt added
Browse files
app.py
CHANGED
@@ -42,12 +42,6 @@ base = "stabilityai/stable-diffusion-xl-base-1.0"
|
|
42 |
repo = "ByteDance/SDXL-Lightning"
|
43 |
ckpt = "sdxl_lightning_4step_unet.safetensors" # Use the correct ckpt for your step setting!
|
44 |
|
45 |
-
# unet = UNet2DConditionModel.from_config(base, subfolder="unet").to("cuda", torch.float16)
|
46 |
-
# unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device="cuda"))
|
47 |
-
# pipe = StableDiffusionXLPipeline.from_pretrained(base, unet=unet, torch_dtype=torch.float16, variant="fp16")
|
48 |
-
# pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing")
|
49 |
-
# pipe = pipe.to(device)
|
50 |
-
|
51 |
|
52 |
# Load model.
|
53 |
unet = UNet2DConditionModel.from_config(base, subfolder="unet").to(device)
|
@@ -168,13 +162,13 @@ with gr.Blocks(css="footer{display:none !important}", theme=theme) as demo:
|
|
168 |
# result = gr.Gallery(label="Left is Base and Right is Lora"),
|
169 |
with gr.Accordion("Advanced options", open=False):
|
170 |
with gr.Row():
|
171 |
-
use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=
|
172 |
use_prompt_2 = gr.Checkbox(label="Use prompt 2", value=False)
|
173 |
use_negative_prompt_2 = gr.Checkbox(label="Use negative prompt 2", value=False)
|
174 |
negative_prompt = gr.Text(
|
175 |
label="Negative prompt",
|
176 |
max_lines=1,
|
177 |
-
placeholder="
|
178 |
visible=False,
|
179 |
)
|
180 |
prompt_2 = gr.Text(
|
|
|
42 |
repo = "ByteDance/SDXL-Lightning"
|
43 |
ckpt = "sdxl_lightning_4step_unet.safetensors" # Use the correct ckpt for your step setting!
|
44 |
|
|
|
|
|
|
|
|
|
|
|
|
|
45 |
|
46 |
# Load model.
|
47 |
unet = UNet2DConditionModel.from_config(base, subfolder="unet").to(device)
|
|
|
162 |
# result = gr.Gallery(label="Left is Base and Right is Lora"),
|
163 |
with gr.Accordion("Advanced options", open=False):
|
164 |
with gr.Row():
|
165 |
+
use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)
|
166 |
use_prompt_2 = gr.Checkbox(label="Use prompt 2", value=False)
|
167 |
use_negative_prompt_2 = gr.Checkbox(label="Use negative prompt 2", value=False)
|
168 |
negative_prompt = gr.Text(
|
169 |
label="Negative prompt",
|
170 |
max_lines=1,
|
171 |
+
placeholder="blur, cartoon, bad, face, painting",
|
172 |
visible=False,
|
173 |
)
|
174 |
prompt_2 = gr.Text(
|