Move model to device to avoid error

examples/zero_shot.py

@@ -272,6 +272,7 @@ if __name__ == "__main__":
     model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(
         "hf-hub:imageomics/bioclip"
     )
+    model = model.to(args.device)
 
     # Write datasets
     params_file = os.path.join(args.logs, args.exp, "params.json")