kadirnar commited on
Commit
fabf8b5
1 Parent(s): 77b6bae

Update diffusion_webui/diffusion_models/text2img_app.py

Browse files
diffusion_webui/diffusion_models/text2img_app.py CHANGED
@@ -20,7 +20,7 @@ class StableDiffusionText2ImageGenerator:
20
  ):
21
  if self.pipe is None or self.pipe.model_name != stable_model_path or self.pipe.scheduler_name != scheduler:
22
  self.pipe = StableDiffusionPipeline.from_pretrained(
23
- model_path, safety_checker=None, torch_dtype=torch.float16
24
  )
25
 
26
  self.pipe = get_scheduler(pipe=self.pipe, scheduler=scheduler)
 
20
  ):
21
  if self.pipe is None or self.pipe.model_name != stable_model_path or self.pipe.scheduler_name != scheduler:
22
  self.pipe = StableDiffusionPipeline.from_pretrained(
23
+ stable_model_path, safety_checker=None, torch_dtype=torch.float16
24
  )
25
 
26
  self.pipe = get_scheduler(pipe=self.pipe, scheduler=scheduler)