Manjushri commited on
Commit
36b6742
1 Parent(s): 5836ac0

Update app.py

Browse files

Removing SDXL VAE.... Not good for this particular model perhaps

Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -4,14 +4,11 @@ import numpy as np
4
  import modin.pandas as pd
5
  from PIL import Image
6
  from diffusers import DiffusionPipeline, StableDiffusionLatentUpscalePipeline
7
- from diffusers.models import AutoencoderKL
8
 
9
  device = "cuda" if torch.cuda.is_available() else "cpu"
10
 
11
- vae = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None)
12
-
13
- pipe = DiffusionPipeline.from_pretrained("dreamlike-art/dreamlike-photoreal-2.0", torch_dtype=torch.float16, safety_checker=None, vae=vae)
14
- upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16, safety_checker=None, vae=vae)
15
  upscaler = upscaler.to(device)
16
  pipe = pipe.to(device)
17
 
 
4
  import modin.pandas as pd
5
  from PIL import Image
6
  from diffusers import DiffusionPipeline, StableDiffusionLatentUpscalePipeline
 
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
+ pipe = DiffusionPipeline.from_pretrained("dreamlike-art/dreamlike-photoreal-2.0", torch_dtype=torch.float16, safety_checker=None)
11
+ upscaler = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16, safety_checker=None)
 
 
12
  upscaler = upscaler.to(device)
13
  pipe = pipe.to(device)
14