try fix preds gpu

embeddings/utils/lightning_callbacks/custom_prediction_writer.py

@@ -1,6 +1,6 @@
 import os
 import pickle
-import torch
+
 from pytorch_lightning.callbacks import BasePredictionWriter
 
 

embeddings/utils/utils.py

@@ -3,11 +3,11 @@
 import os.path
 import pprint
 import zipfile
+from collections.abc import Iterable
 from datetime import datetime
 from pathlib import Path
 from tempfile import NamedTemporaryFile
 from typing import Any, Dict, List, Optional, Tuple, Union
-from collections.abc import Iterable
 
 import numpy as np
 import pkg_resources