Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -26,10 +26,5 @@ def generate(prompt, steps):
|
|
26 |
output_image = gr.Image(type="pil")
|
27 |
demo = gr.Interface(fn=generate, inputs=[gr.Text, gr.slider], outputs=output_image)
|
28 |
|
29 |
-
|
30 |
-
|
31 |
-
unet.load_state_dict(load_file(hf_hub_download(repo, ckpt), device="cuda"))
|
32 |
-
pipe = StableDiffusionXLPipeline.from_pretrained(base, unet=unet, torch_dtype=torch.float16, variant="fp16").to("cuda")
|
33 |
-
# Ensure sampler uses "trailing" timesteps and "sample" prediction type.
|
34 |
-
pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, timestep_spacing="trailing", prediction_type="sample")
|
35 |
-
demo.launch()
|
|
|
26 |
output_image = gr.Image(type="pil")
|
27 |
demo = gr.Interface(fn=generate, inputs=[gr.Text, gr.slider], outputs=output_image)
|
28 |
|
29 |
+
|
30 |
+
demo.launch()
|
|
|
|
|
|
|
|
|
|