From 457d1561a1b99eadbac64fb7593c1d456ffcf1c8 Mon Sep 17 00:00:00 2001
From: matt3o <matt3o@gmail.com>
Date: Sun, 24 Mar 2024 00:00:46 +0100
Subject: [PATCH] fallback for old torch versions

---
 utils.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/utils.py b/utils.py
index 3d3b942..e2e3ffc 100644
--- a/utils.py
+++ b/utils.py
@@ -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()