add back enable_attention_slicing for mps device

2024-01-08 21:49:18 +08:00
parent 3b40671e33
commit 5da47ee035
7 changed files with 20 additions and 2 deletions
--- a/iopaint/model/utils.py
+++ b/iopaint/model/utils.py
@@ -24,7 +24,6 @@ from diffusers import (
    KDPM2AncestralDiscreteScheduler,
    HeunDiscreteScheduler,
 )
-from diffusers.configuration_utils import FrozenDict
 from loguru import logger

 from iopaint.schema import SDSampler