Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
45d444d
1
Parent(s):
c8041e9
add inference
Browse files
app.py
CHANGED
|
@@ -161,7 +161,7 @@ repo_id = "madebyollin/sdxl-vae-fp16-fix" # e.g., "distilbert/distilgpt2"
|
|
| 161 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix",torch_dtype=torch_dtype) #from_single_file(downloaded_path, torch_dtype=torch_dtype)
|
| 162 |
vae.to('cuda')
|
| 163 |
|
| 164 |
-
pipe = StableDiffusionXLPipeline.from_pretrained("
|
| 165 |
# pipe = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0",torch_dtype=torch.float16,vae=vae)
|
| 166 |
|
| 167 |
pipe.to('cuda')
|
|
|
|
| 161 |
vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix",torch_dtype=torch_dtype) #from_single_file(downloaded_path, torch_dtype=torch_dtype)
|
| 162 |
vae.to('cuda')
|
| 163 |
|
| 164 |
+
pipe = StableDiffusionXLPipeline.from_pretrained("John6666/nova-anime-xl-il-v120-sdxl",torch_dtype=torch_dtype,vae=vae)
|
| 165 |
# pipe = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0",torch_dtype=torch.float16,vae=vae)
|
| 166 |
|
| 167 |
pipe.to('cuda')
|