Kkordik commited on
Commit
44eac71
1 Parent(s): 49cba59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -5,8 +5,10 @@ import torch
5
  from PIL import Image, ImageOps
6
  import time
7
 
8
- vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
9
- pipeline = StableDiffusion3InpaintPipeline(vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to("cuda")
 
 
10
 
11
  def get_select_index(evt: gr.SelectData):
12
  return evt.index
 
5
  from PIL import Image, ImageOps
6
  import time
7
 
8
+ # vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
9
+ # pipeline = StableDiffusion3InpaintPipeline(vae=vae, torch_dtype=torch.float16, variant="fp16", use_safetensors=True).to("cuda")
10
+
11
+ pipeline = StableDiffusion3InpaintPipeline()
12
 
13
  def get_select_index(evt: gr.SelectData):
14
  return evt.index