fix test

2022-12-11 22:03:36 +08:00
parent ad8437dc9c
commit 221d44c037
3 changed files with 2 additions and 2 deletions
--- a/lama_cleaner/model/sd.py
+++ b/lama_cleaner/model/sd.py
@@ -50,7 +50,7 @@ class SD(InpaintModel):
        # https://huggingface.co/docs/diffusers/v0.7.0/en/api/pipelines/stable_diffusion#diffusers.StableDiffusionInpaintPipeline.enable_attention_slicing
        self.model.enable_attention_slicing()
        # https://huggingface.co/docs/diffusers/v0.7.0/en/optimization/fp16#memory-efficient-attention
-        if kwargs['sd_enable_xformers']:
+        if kwargs.get('sd_enable_xformers', False):
            self.model.enable_xformers_memory_efficient_attention()
        self.model = self.model.to(device)