torch_vision.py: use set device type (#621)

* use device enum to transfer model to cuda

* Update libs/infinity_emb/infinity_emb/transformer/vision/torch_vision.py

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

---------

Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>

Author: wirthual

libs/infinity_emb/infinity_emb/transformer/vision/torch_vision.py

@@ -106,7 +106,7 @@ def __init__(self, *, engine_args: "EngineArgs"):
             assert hasattr(
                 self.model, "get_image_features"
             ), f"AutoModel of {engine_args.model_name_or_path} does not have get_image_features method"
-        if torch.cuda.is_available():
+        if device == Device.cuda:
             self.model = self.model.cuda()
             if engine_args.dtype in (Dtype.float16, Dtype.auto):
                 self.model = self.model.half()