Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,6 @@ state_dict = load_state_dict(model_file)
|
|
| 30 |
model, _, _, _, _ = ControlNetModel_Union._load_pretrained_model(
|
| 31 |
controlnet_model, state_dict, model_file, "xinsir/controlnet-union-sdxl-1.0"
|
| 32 |
)
|
| 33 |
-
model.to(device="cpu", dtype=torch.float16)
|
| 34 |
|
| 35 |
vae = AutoencoderKL.from_pretrained(
|
| 36 |
"madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
|
|
@@ -276,15 +275,6 @@ with gr.Blocks(css=css) as demo:
|
|
| 276 |
|
| 277 |
run_button = gr.Button("Generate")
|
| 278 |
|
| 279 |
-
gr.Examples(
|
| 280 |
-
examples = [
|
| 281 |
-
["./examples/example_1.webp", "RealVisXL V5.0 Lightning", "16:9"],
|
| 282 |
-
["./examples/example_2.jpg", "RealVisXL V5.0 Lightning", "16:9"],
|
| 283 |
-
["./examples/example_3.jpg", "RealVisXL V5.0 Lightning", "9:16"]
|
| 284 |
-
],
|
| 285 |
-
inputs = [input_image, model_selection, ratio]
|
| 286 |
-
)
|
| 287 |
-
|
| 288 |
with gr.Column():
|
| 289 |
result = ImageSlider(
|
| 290 |
interactive=False,
|
|
@@ -302,4 +292,4 @@ with gr.Blocks(css=css) as demo:
|
|
| 302 |
)
|
| 303 |
|
| 304 |
|
| 305 |
-
demo.launch(
|
|
|
|
| 30 |
model, _, _, _, _ = ControlNetModel_Union._load_pretrained_model(
|
| 31 |
controlnet_model, state_dict, model_file, "xinsir/controlnet-union-sdxl-1.0"
|
| 32 |
)
|
|
|
|
| 33 |
|
| 34 |
vae = AutoencoderKL.from_pretrained(
|
| 35 |
"madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16
|
|
|
|
| 275 |
|
| 276 |
run_button = gr.Button("Generate")
|
| 277 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 278 |
with gr.Column():
|
| 279 |
result = ImageSlider(
|
| 280 |
interactive=False,
|
|
|
|
| 292 |
)
|
| 293 |
|
| 294 |
|
| 295 |
+
demo.launch()
|