UdacityNoob commited on
Commit
96a1a23
β€’
1 Parent(s): cf11c5a

Un-comment code

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ TOKEN_KEY = os.getenv('AUTH_TOKEN')
9
 
10
  # setup pipeline
11
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=TOKEN_KEY)
12
- # pipe = pipe.to('cuda')
13
 
14
  # define gradio function
15
  def generate(prompt:str, seed:int, guidance:float):
 
9
 
10
  # setup pipeline
11
  pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", revision="fp16", torch_dtype=torch.float16, use_auth_token=TOKEN_KEY)
12
+ pipe = pipe.to('cuda')
13
 
14
  # define gradio function
15
  def generate(prompt:str, seed:int, guidance:float):