Fix filtered Comfy dependencies

app.py

@@ -6,6 +6,7 @@ import json
 import os
 import pathlib
 import random
+import re
 import shutil
 import subprocess
 import sys
@@ -121,14 +122,15 @@ def _pip_install(args: list[str], check: bool = True) -> None:
 def _install_filtered_requirements(req_path: pathlib.Path) -> None:
     if not req_path.exists():
         return
-    blocked = ("torch", "torchvision", "torchaudio", "transformers", "huggingface-hub", "accelerate")
+    blocked = {"torch", "torchvision", "torchaudio", "transformers", "huggingface-hub", "accelerate"}
     safe: list[str] = []
     for line in req_path.read_text(encoding="utf-8", errors="ignore").splitlines():
         item = line.strip()
         if not item or item.startswith("#"):
             continue
         low = item.lower().replace("_", "-")
-        if any(low.startswith(name) for name in blocked):
+        package = re.split(r"[<>=!~;\[\s]", low, maxsplit=1)[0]
+        if package in blocked:
             continue
         safe.append(item)
     if safe:

requirements.txt

@@ -2,6 +2,7 @@
 torch==2.8.0
 torchvision==0.23.0
 torchaudio==2.8.0
+torchsde
 spaces
 gradio>=5,<6
 huggingface_hub>=0.34.0