Update model.py

2025-05-14 10:31:44 +03:00
parent bfed2afc2e
commit abdac5ec3a
1 changed files with 1 additions and 0 deletions
@@ -14,6 +14,7 @@ try:
    create_block_mask = torch.compile(create_block_mask)
    flex_attention = torch.compile(flex_attention)
 except:
+    BlockMask = create_block_mask = flex_attention = None
    pass

 from .attention import attention