From d9f85fcf1e4024bfb64a6f009be4d66336cacfe1 Mon Sep 17 00:00:00 2001
From: "DESKTOP-UL3K6JV\\Daniq" <31.10.93@bk.ru>
Date: Sun, 16 Jun 2024 17:09:24 +0500
Subject: [PATCH] Changed permute

---
 classifiers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/classifiers.py b/classifiers.py
index 433e72e..0eb227a 100644
--- a/classifiers.py
+++ b/classifiers.py
@@ -55,7 +55,7 @@ class YOLOClassify:
     CATEGORY = "SuperMasterBlasterLaser/ComfyUI_YOLO_Classifiers"
 
     def classify(self, yolo_classifier_model, image_to_classify):
-        image_to_classify = image_to_classify.permute((0, 3, 1, 2))
+        image_to_classify = image_to_classify.permute(0, 3, 1, 2)
 
         transform = transforms.Compose([
             transforms.Resize(size=(640, 640))