chansung commited on
Commit
08ed65e
·
1 Parent(s): 37d432e

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -0
handler.py CHANGED
@@ -10,6 +10,7 @@ from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
10
 
11
  class EndpointHandler():
12
  def __init__(self, path="", model_base="runwayml/stable-diffusion-v1-5"):
 
13
  self.pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16)
14
  self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(self.pipe.scheduler.config)
15
  self.original_unet = copy.deepcopy(self.pipe.unet)
 
10
 
11
  class EndpointHandler():
12
  def __init__(self, path="", model_base="runwayml/stable-diffusion-v1-5"):
13
+ print("diffusers.__version__: " + diffusers.__version__)
14
  self.pipe = StableDiffusionPipeline.from_pretrained(model_base, torch_dtype=torch.float16)
15
  self.pipe.scheduler = DPMSolverMultistepScheduler.from_config(self.pipe.scheduler.config)
16
  self.original_unet = copy.deepcopy(self.pipe.unet)