Fix balanced offload with Cascade

This commit is contained in:
Disty0
2024-12-13 23:01:53 +03:00
parent 8ee5103ade
commit 7d7bcb9684
2 changed files with 3 additions and 1 deletions
+1 -1
View File
@@ -1,7 +1,7 @@
import os
from functools import wraps, cache
import torch
import diffusers #0.29.1 # pylint: disable=import-error
import diffusers # pylint: disable=import-error
from diffusers.models.attention_processor import Attention
# pylint: disable=protected-access, missing-function-docstring, line-too-long