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

Will work? Using Inference API

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ def greet(name):
13
  def create_waifu(prompt):
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]
 
13
  def create_waifu(prompt):
14
  pipe = StableDiffusionPipeline.from_pretrained(
15
  'hakurei/waifu-diffusion',
16
+ torch_dtype=torch.float16
17
  )
18
 
19
  image = pipe(prompt, guidance_scale=6)[0][0]