Update ovis_image/test.py

ovis_image/test.py

@@ -38,7 +38,7 @@ def load_model_weight(model, model_path):
 def main():
     args = parse_args()
     model_config = ovis_image_configs["ovis-image-7b"]
-    device = "cuda"
+    device = "cuda" if torch.cuda.is_available() else "cpu"
     _dtype = torch.bfloat16
     print(f"dtype: {_dtype}")
     ovis_image = OvisImageModel(model_config)