apply lint
This commit is contained in:
parent
f534fbb92c
commit
0d2211ff25
|
@ -121,7 +121,6 @@ def optimize_pipeline(
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
logger.warning("error while enabling model CPU offload: %s", e)
|
logger.warning("error while enabling model CPU offload: %s", e)
|
||||||
|
|
||||||
|
|
||||||
if "memory-efficient-attention" in server.optimizations:
|
if "memory-efficient-attention" in server.optimizations:
|
||||||
# TODO: check for xformers
|
# TODO: check for xformers
|
||||||
logger.debug("enabling memory efficient attention for SD pipeline")
|
logger.debug("enabling memory efficient attention for SD pipeline")
|
||||||
|
|
Loading…
Reference in New Issue