kadirnar commited on
Commit
a2754d8
1 Parent(s): 6116da7

Update diffusion_webui/diffusion_models/img2img_app.py

Browse files
diffusion_webui/diffusion_models/img2img_app.py CHANGED
@@ -17,7 +17,7 @@ class StableDiffusionImage2ImageGenerator:
17
  def load_model(self, stable_model_path, scheduler):
18
  if self.pipe is None or self.pipe.model_name != stable_model_path or self.pipe.scheduler_name != scheduler:
19
  self.pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
20
- model_path, safety_checker=None, torch_dtype=torch.float16
21
  )
22
 
23
  self.pipe.model_name = stable_model_path
 
17
  def load_model(self, stable_model_path, scheduler):
18
  if self.pipe is None or self.pipe.model_name != stable_model_path or self.pipe.scheduler_name != scheduler:
19
  self.pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
20
+ stable_model_path, safety_checker=None, torch_dtype=torch.float16
21
  )
22
 
23
  self.pipe.model_name = stable_model_path