Will-uob commited on
Commit
729ee01
·
1 Parent(s): 5471374

Got rid of cuda requirement. Really this is a major problem with this idea.

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -14,10 +14,9 @@ def create_waifu(prompt):
14
  pipe = StableDiffusionPipeline.from_pretrained(
15
  'hakurei/waifu-diffusion',
16
  torch_dtype=torch.float32
17
- ).to('cuda')
18
 
19
- with autocast("cuda"):
20
- image = pipe(prompt, guidance_scale=6)[0][0]
21
 
22
  return image
23
  """
 
14
  pipe = StableDiffusionPipeline.from_pretrained(
15
  'hakurei/waifu-diffusion',
16
  torch_dtype=torch.float32
17
+ )
18
 
19
+ image = pipe(prompt, guidance_scale=6)[0][0]
 
20
 
21
  return image
22
  """