Hermes Bot commited on
Commit
03a3e87
·
1 Parent(s): bec4723

Disable SageAttention to avoid CPU errors

Browse files
app.py CHANGED
@@ -23,13 +23,13 @@ def apply_sage_attention_patch():
23
  from comfy import model_management
24
  import sageattention
25
  print("--- [Runtime Patch] sageattention package found. Applying patch... ---")
26
- # Safely set flags if they exist; on CPU-only setups model_management may be limited.
27
  if hasattr(model_management, "sage_attention_enabled"):
28
- model_management.sage_attention_enabled = lambda: True
29
  if hasattr(model_management, "pytorch_attention_enabled"):
30
- model_management.pytorch_attention_enabled = lambda: False
31
  SAGE_PATCH_APPLIED = True
32
- return "✅ Successfully enabled SageAttention."
33
  except Exception as e:
34
  SAGE_PATCH_APPLIED = False
35
  msg = f"--- [Runtime Patch] ⚠️ SageAttention patch could not be applied: {e} ---"
 
23
  from comfy import model_management
24
  import sageattention
25
  print("--- [Runtime Patch] sageattention package found. Applying patch... ---")
26
+ # Disable SageAttention to avoid runtime errors on CPUonly setups.
27
  if hasattr(model_management, "sage_attention_enabled"):
28
+ model_management.sage_attention_enabled = lambda: False
29
  if hasattr(model_management, "pytorch_attention_enabled"):
30
+ model_management.pytorch_attention_enabled = lambda: True
31
  SAGE_PATCH_APPLIED = True
32
+ return "✅ Successfully disabled SageAttention (using standard PyTorch attention)."
33
  except Exception as e:
34
  SAGE_PATCH_APPLIED = False
35
  msg = f"--- [Runtime Patch] ⚠️ SageAttention patch could not be applied: {e} ---"
comfy_integration/__pycache__/setup.cpython-311.pyc CHANGED
Binary files a/comfy_integration/__pycache__/setup.cpython-311.pyc and b/comfy_integration/__pycache__/setup.cpython-311.pyc differ
 
requirements.txt CHANGED
@@ -1,46 +1,46 @@
1
- comfyui-frontend-package==1.45.20
2
- comfyui-workflow-templates==0.11.1
3
- comfyui-embedded-docs==0.5.6
4
- torch
5
- torchsde
6
- torchvision
7
- torchaudio
8
- numpy>=1.25.0
9
- einops
10
- transformers>=4.50.3
11
- tokenizers>=0.13.3
12
- sentencepiece
13
- safetensors>=0.4.2
14
- aiohttp>=3.11.8
15
- yarl>=1.18.0
16
- pyyaml
17
- Pillow
18
- scipy
19
- tqdm
20
- psutil
21
- alembic
22
- SQLAlchemy>=2.0.0
23
- filelock
24
- av>=16.0.0
25
- comfy-kitchen==0.2.16
26
- comfy-aimdo==0.4.10
27
- requests
28
- simpleeval>=1.0.0
29
- blake3
30
-
31
- #non essential dependencies:
32
- kornia>=0.7.1
33
- spandrel
34
- pydantic~=2.0
35
- pydantic-settings~=2.0
36
- PyOpenGL>=3.1.8
37
- comfy-angle
38
-
39
-
40
- diffusers
41
- protobuf
42
- insightface
43
- huggingface-hub
44
- imageio
45
- spaces
46
- sageattention @ https://huggingface.co/RioShiina/Sage-Attention-ZeroGPU-Space-Build/resolve/main/sageattention-2.2.0-cp312-cp312-linux_x86_64.whl
 
1
+ comfyui-frontend-package==1.45.20
2
+ comfyui-workflow-templates==0.11.1
3
+ comfyui-embedded-docs==0.5.6
4
+ # torch
5
+ # torchsde
6
+ # torchvision
7
+ # torchaudio
8
+ numpy>=1.25.0
9
+ einops
10
+ transformers>=4.50.3
11
+ tokenizers>=0.13.3
12
+ sentencepiece
13
+ safetensors>=0.4.2
14
+ aiohttp>=3.11.8
15
+ yarl>=1.18.0
16
+ pyyaml
17
+ Pillow
18
+ scipy
19
+ tqdm
20
+ psutil
21
+ alembic
22
+ SQLAlchemy>=2.0.0
23
+ filelock
24
+ av>=16.0.0
25
+ comfy-kitchen==0.2.16
26
+ comfy-aimdo==0.4.10
27
+ requests
28
+ simpleeval>=1.0.0
29
+ blake3
30
+
31
+ #non essential dependencies:
32
+ kornia>=0.7.1
33
+ spandrel
34
+ pydantic~=2.0
35
+ pydantic-settings~=2.0
36
+ PyOpenGL>=3.1.8
37
+ comfy-angle
38
+
39
+
40
+ diffusers
41
+ protobuf
42
+ insightface
43
+ huggingface-hub
44
+ imageio
45
+ spaces
46
+ # sageattention @ https://huggingface.co/RioShiina/Sage-Attention-ZeroGPU-Space-Build/resolve/main/sageattention-2.2.0-cp312-cp312-linux_x86_64.whl