guardiancc
commited on
Commit
•
17c4e15
1
Parent(s):
061cbe1
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()
|