fallback for old torch versions

2024-03-24 00:00:46 +01:00
parent 586854b0ba
commit 457d1561a1
1 changed files with 4 additions and 1 deletions
@@ -5,7 +5,10 @@ import folder_paths
 from comfy.clip_vision import clip_preprocess, Output
 import comfy.utils
 import comfy.model_management as model_management
-import torchvision.transforms.v2 as T
+try:
+    import torchvision.transforms.v2 as T
+except ImportError:
+    import torchvision.transforms as T

 def get_clipvision_file(preset):
    preset = preset.lower()