fffiloni commited on
Commit
4c97218
1 Parent(s): fe22cdf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -14,7 +14,6 @@ pipe = DiffusionPipeline.from_pretrained(
14
  safety_checker=None,
15
  use_auth_token=MY_SECRET_TOKEN,
16
  revision="fp16",
17
- torch_dtype=torch.float16,
18
  custom_pipeline="imagic_stable_diffusion",
19
  scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
20
  ).to(device)
@@ -32,6 +31,8 @@ def infer(prompt, init_image):
32
  num_inference_steps=50,
33
  generator=generator)
34
 
 
 
35
  res = pipe(alpha=1)
36
 
37
  return res.images[0]
 
14
  safety_checker=None,
15
  use_auth_token=MY_SECRET_TOKEN,
16
  revision="fp16",
 
17
  custom_pipeline="imagic_stable_diffusion",
18
  scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
19
  ).to(device)
 
31
  num_inference_steps=50,
32
  generator=generator)
33
 
34
+ torch.cuda.empty_cache()
35
+
36
  res = pipe(alpha=1)
37
 
38
  return res.images[0]