mochifz commited on
Commit
04c85ef
1 Parent(s): 1d716f9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
  import torch, random, time
3
- from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, EulerDiscreteScheduler
4
  device = "cuda" if torch.cuda.is_available() else "cpu"
5
  translations = {
6
  'en': {
@@ -74,7 +74,6 @@ def Loading(model):
74
  else:
75
  text2img = StableDiffusionPipeline.from_pretrained(model, use_safetensors=True).to(device)
76
  text2img.safety_checker = None
77
- text2img.scheduler = EulerDiscreteScheduler.from_config(text2img.scheduler.config)
78
  img2img = StableDiffusionImg2ImgPipeline(**text2img.components)
79
  return model
80
  with gr.Blocks() as demo:
 
1
  import gradio as gr
2
  import torch, random, time
3
+ from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
4
  device = "cuda" if torch.cuda.is_available() else "cpu"
5
  translations = {
6
  'en': {
 
74
  else:
75
  text2img = StableDiffusionPipeline.from_pretrained(model, use_safetensors=True).to(device)
76
  text2img.safety_checker = None
 
77
  img2img = StableDiffusionImg2ImgPipeline(**text2img.components)
78
  return model
79
  with gr.Blocks() as demo: