albarji commited on
Commit
c9faa86
1 Parent(s): 6320655

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from mixdiff import StableDiffusionCanvasPipeline, Text2ImageRegion
5
 
6
  # Creater scheduler and model (similar to StableDiffusionPipeline)
7
  scheduler = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000)
8
- pipeline = StableDiffusionCanvasPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", scheduler=scheduler, use_auth_token=True).to("cuda:0")
9
 
10
  def generate(prompt1, prompt2, prompt3, seed):
11
  """Mixture of Diffusers generation"""
 
5
 
6
  # Creater scheduler and model (similar to StableDiffusionPipeline)
7
  scheduler = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000)
8
+ pipeline = StableDiffusionCanvasPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", scheduler=scheduler).to("cuda:0")
9
 
10
  def generate(prompt1, prompt2, prompt3, seed):
11
  """Mixture of Diffusers generation"""