fixed causvid scheduler
This commit is contained in:
@@ -23,14 +23,7 @@ from .mlp import Mlp
|
||||
logger = logging.getLogger("dinov2")
|
||||
|
||||
|
||||
try:
|
||||
from xformers.ops import fmha
|
||||
from xformers.ops import scaled_index_add, index_select_cat
|
||||
|
||||
XFORMERS_AVAILABLE = True
|
||||
except ImportError:
|
||||
# logger.warning("xFormers not available")
|
||||
XFORMERS_AVAILABLE = False
|
||||
XFORMERS_AVAILABLE = False
|
||||
|
||||
|
||||
class Block(nn.Module):
|
||||
|
||||
Reference in New Issue
Block a user