apruvd commited on
Commit
12bac1c
·
1 Parent(s): 0a9ef54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ scheduler = EulerDiscreteScheduler.from_pretrained(model_id, subfolder="schedule
16
  pipe = StableDiffusionPipeline.from_pretrained(model_id , torch_dtype=torch.float16) #pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
17
  pipe = pipe.to("cuda")
18
 
19
- from IPython.display import Image
20
  from PIL import Image
21
  import time
22
  import matplotlib.pyplot as plt
 
16
  pipe = StableDiffusionPipeline.from_pretrained(model_id , torch_dtype=torch.float16) #pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler, revision="fp16", torch_dtype=torch.float16)
17
  pipe = pipe.to("cuda")
18
 
19
+ # from IPython.display import Image
20
  from PIL import Image
21
  import time
22
  import matplotlib.pyplot as plt