noamelata commited on
Commit
dbcbdf5
·
1 Parent(s): 9d761af

changed to full precision

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -34,8 +34,12 @@ def run(prompt, outer, inner, random_seed, pipe):
34
  if __name__ == "__main__":
35
  scheduler = NestedScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear",
36
  prediction_type='sample', clip_sample=False, set_alpha_to_one=False)
37
- pipe = NestedStableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", revision="fp16",
 
 
38
  torch_dtype=torch.float16, scheduler=scheduler)
 
 
39
  device = "cuda" if torch.cuda.is_available() else "cpu"
40
  pipe.to(device)
41
  interface = partial(run, pipe=pipe)
 
34
  if __name__ == "__main__":
35
  scheduler = NestedScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear",
36
  prediction_type='sample', clip_sample=False, set_alpha_to_one=False)
37
+ fp16 = False
38
+ if fp16:
39
+ pipe = NestedStableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", revision="fp16",
40
  torch_dtype=torch.float16, scheduler=scheduler)
41
+ else:
42
+ pipe = NestedStableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", scheduler=scheduler)
43
  device = "cuda" if torch.cuda.is_available() else "cpu"
44
  pipe.to(device)
45
  interface = partial(run, pipe=pipe)