Samuel L Meyers
commited on
Commit
·
55db009
1
Parent(s):
737026e
Meow
Browse files- code/app.py +1 -1
code/app.py
CHANGED
|
@@ -7,7 +7,7 @@ vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype
|
|
| 7 |
# "stabilityai/stable-diffusion-xl-base-1.0", use_safetensors=True, revision="fp16", torch_dtype=torch.float16, vae=vae
|
| 8 |
# )
|
| 9 |
pipeline_text2image = DiffusionPipeline.from_pretrained(
|
| 10 |
-
"ARDICAI/stable-diffusion-2-1-finetuned", use_safetensors=True,
|
| 11 |
)
|
| 12 |
|
| 13 |
def gradio_txt2img(prompt):
|
|
|
|
| 7 |
# "stabilityai/stable-diffusion-xl-base-1.0", use_safetensors=True, revision="fp16", torch_dtype=torch.float16, vae=vae
|
| 8 |
# )
|
| 9 |
pipeline_text2image = DiffusionPipeline.from_pretrained(
|
| 10 |
+
"ARDICAI/stable-diffusion-2-1-finetuned", use_safetensors=True, vae=vae
|
| 11 |
)
|
| 12 |
|
| 13 |
def gradio_txt2img(prompt):
|