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

Update diffusion_webui/diffusion_models/inpaint_app.py

Browse files
diffusion_webui/diffusion_models/inpaint_app.py CHANGED
@@ -12,7 +12,7 @@ class StableDiffusionInpaintGenerator:
12
  def load_model(self, stable_model_path):
13
  if self.pipe is None or self.pipe.model_name != stable_model_path:
14
  self.pipe = DiffusionPipeline.from_pretrained(
15
- model_path, revision="fp16", torch_dtype=torch.float16
16
  )
17
  self.pipe.to("cuda")
18
  self.pipe.enable_xformers_memory_efficient_attention()
 
12
  def load_model(self, stable_model_path):
13
  if self.pipe is None or self.pipe.model_name != stable_model_path:
14
  self.pipe = DiffusionPipeline.from_pretrained(
15
+ stable_model_path, revision="fp16", torch_dtype=torch.float16
16
  )
17
  self.pipe.to("cuda")
18
  self.pipe.enable_xformers_memory_efficient_attention()