From efa350aba4194e53382559f532a60b7e9a1abac6 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sun, 18 Feb 2024 16:38:53 +0200
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index 03419c3..da94fa6 100644
--- a/nodes.py
+++ b/nodes.py
@@ -3639,7 +3639,7 @@ class EffnetEncode:
 
         effnet.load_state_dict(effnet_state_dict if 'state_dict' not in effnet_checkpoint else effnet_checkpoint['state_dict'])
         effnet.eval().requires_grad_(False).to(device)
-        t = effnet(image)
+        t = effnet(image).cpu()
         return ({"samples":t}, )