Manjushri commited on
Commit
13f55b4
1 Parent(s): 3fa1235

Update app.py

Browse files

Trying default scheduler

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -3,12 +3,12 @@ import torch
3
  import numpy as np
4
  import modin.pandas as pd
5
  from PIL import Image
6
- from diffusers import DiffusionPipeline, EulerDiscreteScheduler
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
- scheduler = EulerDiscreteScheduler.from_pretrained("stabilityai/stable-diffusion-2-1", subfolder="scheduler", prediction_type="v_prediction")
11
- pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", scheduler=scheduler, safety_checker=None)
12
  pipe = pipe.to(device)
13
 
14
  def genie (prompt, negative_prompt, scale, steps, seed):
 
3
  import numpy as np
4
  import modin.pandas as pd
5
  from PIL import Image
6
+ from diffusers import DiffusionPipeline, #EulerDiscreteScheduler
7
 
8
  device = "cuda" if torch.cuda.is_available() else "cpu"
9
 
10
+ #scheduler = EulerDiscreteScheduler.from_pretrained("stabilityai/stable-diffusion-2-1", subfolder="scheduler", prediction_type="v_prediction")
11
+ pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-2-1", safety_checker=None)
12
  pipe = pipe.to(device)
13
 
14
  def genie (prompt, negative_prompt, scale, steps, seed):