Manjushri commited on
Commit
9d9dad1
1 Parent(s): bcde413

Update app.py

Browse files

Upgraded to SDXL 1.0

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -5,12 +5,12 @@ import numpy as np
5
  from PIL import Image
6
  from diffusers import DiffusionPipeline
7
  from huggingface_hub import login
8
- import os
9
 
10
- login(token=os.environ.get('HF_KEY'))
11
 
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
- pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9")
14
  pipe = pipe.to(device)
15
 
16
  def resize(value,img):
 
5
  from PIL import Image
6
  from diffusers import DiffusionPipeline
7
  from huggingface_hub import login
8
+ #import os
9
 
10
+ #login(token=os.environ.get('HF_KEY'))
11
 
12
  device = "cuda" if torch.cuda.is_available() else "cpu"
13
+ pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtype=torch.float16) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0")
14
  pipe = pipe.to(device)
15
 
16
  def resize(value,img):