Thafx commited on
Commit
abd6f83
1 Parent(s): e8d33c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -12,11 +12,6 @@ pipe = StableDiffusionPipeline.from_pretrained(
12
  model_id,
13
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
14
  scheduler=scheduler)
15
-
16
-
17
- pipe = StableDiffusionPipeline.from_pretrained("hakurei/waifu-diffusion", use_auth_token="<mytoken>").to('cuda')
18
- image = pipe(prompt, safety_checker=None).images[0]
19
- image.save(f"{prompt}.png")
20
 
21
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
22
  model_id,
 
12
  model_id,
13
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
14
  scheduler=scheduler)
 
 
 
 
 
15
 
16
  pipe_i2i = StableDiffusionImg2ImgPipeline.from_pretrained(
17
  model_id,