Debug error inside get_embeddings

functions.py

@@ -44,9 +44,10 @@ def get_embeddings(text):
     encoded_input = {k: v.to('cuda') for k, v in encoded_input.items()}
     model_output = emb_model(**encoded_input)
     model_output = model_output.last_hidden_state[:, 0]
-    print(model_output)
-    emb_item = model_output.detach().cpu().numpy()[0]
-    print(emb_item)
+    # print(model_output)
+    # Error: AttributeError: 'numpy.ndarray' object has no attribute 'cpu'
+    # emb_item = model_output.detach().cpu().numpy()[0]
+    # print(emb_item)
     print(['get_embeddings', 'end'])
     return emb_item