Spaces:
Runtime error
Runtime error
gokaygokay
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -77,21 +77,18 @@ pipe_pony = StableDiffusionXLPipeline.from_pretrained(
|
|
77 |
vae=vae_pony,
|
78 |
torch_dtype=torch.float16,
|
79 |
use_safetensors=True,
|
80 |
-
variant="fp16"
|
81 |
)
|
82 |
pipe_cyber = StableDiffusionXLPipeline.from_pretrained(
|
83 |
ckpt_dir_cyber,
|
84 |
vae=vae_cyber,
|
85 |
torch_dtype=torch.float16,
|
86 |
use_safetensors=True,
|
87 |
-
variant="fp16"
|
88 |
)
|
89 |
pipe_stallion = StableDiffusionXLPipeline.from_pretrained(
|
90 |
ckpt_dir_stallion,
|
91 |
vae=vae_stallion,
|
92 |
torch_dtype=torch.float16,
|
93 |
use_safetensors=True,
|
94 |
-
variant="fp16"
|
95 |
)
|
96 |
|
97 |
pipe_pony = pipe_pony.to("cuda")
|
|
|
77 |
vae=vae_pony,
|
78 |
torch_dtype=torch.float16,
|
79 |
use_safetensors=True,
|
|
|
80 |
)
|
81 |
pipe_cyber = StableDiffusionXLPipeline.from_pretrained(
|
82 |
ckpt_dir_cyber,
|
83 |
vae=vae_cyber,
|
84 |
torch_dtype=torch.float16,
|
85 |
use_safetensors=True,
|
|
|
86 |
)
|
87 |
pipe_stallion = StableDiffusionXLPipeline.from_pretrained(
|
88 |
ckpt_dir_stallion,
|
89 |
vae=vae_stallion,
|
90 |
torch_dtype=torch.float16,
|
91 |
use_safetensors=True,
|
|
|
92 |
)
|
93 |
|
94 |
pipe_pony = pipe_pony.to("cuda")
|