cosmicroxks commited on
Commit
4eb5a53
1 Parent(s): 95437d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ import os
12
  pipe = StableDiffusionPipeline.from_pretrained(
13
  "CompVis/stable-diffusion-v1-4",
14
  use_auth_token="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"
15
- ).to(device)
16
 
17
  def infer(prompt, samples, steps, scale, seed):
18
  generator = torch.Generator(device=device).manual_seed(seed)
 
12
  pipe = StableDiffusionPipeline.from_pretrained(
13
  "CompVis/stable-diffusion-v1-4",
14
  use_auth_token="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"
15
+ ).to("device")
16
 
17
  def infer(prompt, samples, steps, scale, seed):
18
  generator = torch.Generator(device=device).manual_seed(seed)