Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -14,7 +14,7 @@ login(token=a )
|
|
| 14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 15 |
#model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like to use
|
| 16 |
#model_repo_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
| 17 |
-
model_repo_id = "stabilityai/stable-diffusion-xl-base-0.9"
|
| 18 |
|
| 19 |
|
| 20 |
|
|
@@ -140,7 +140,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 140 |
label="Negative prompt",
|
| 141 |
max_lines=1,
|
| 142 |
placeholder="Enter a negative prompt",
|
| 143 |
-
visible=
|
| 144 |
)
|
| 145 |
|
| 146 |
seed = gr.Slider(
|
|
@@ -148,7 +148,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 148 |
minimum=0,
|
| 149 |
maximum=MAX_SEED,
|
| 150 |
step=1,
|
| 151 |
-
value=
|
| 152 |
)
|
| 153 |
|
| 154 |
randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
|
|
@@ -166,7 +166,7 @@ with gr.Blocks(css=css) as demo:
|
|
| 166 |
label="Height",
|
| 167 |
minimum=256,
|
| 168 |
maximum=MAX_IMAGE_SIZE,
|
| 169 |
-
step=
|
| 170 |
value=512, # Replace with defaults that work for your model
|
| 171 |
)
|
| 172 |
|
|
|
|
| 14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 15 |
#model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like to use
|
| 16 |
#model_repo_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
| 17 |
+
model_repo_id = "ByteDance/SDXL-Lightning" #"stabilityai/stable-diffusion-xl-base-0.9"
|
| 18 |
|
| 19 |
|
| 20 |
|
|
|
|
| 140 |
label="Negative prompt",
|
| 141 |
max_lines=1,
|
| 142 |
placeholder="Enter a negative prompt",
|
| 143 |
+
visible=False,
|
| 144 |
)
|
| 145 |
|
| 146 |
seed = gr.Slider(
|
|
|
|
| 148 |
minimum=0,
|
| 149 |
maximum=MAX_SEED,
|
| 150 |
step=1,
|
| 151 |
+
value=4,
|
| 152 |
)
|
| 153 |
|
| 154 |
randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
|
|
|
|
| 166 |
label="Height",
|
| 167 |
minimum=256,
|
| 168 |
maximum=MAX_IMAGE_SIZE,
|
| 169 |
+
step=24,
|
| 170 |
value=512, # Replace with defaults that work for your model
|
| 171 |
)
|
| 172 |
|