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}, )