Move model to device to avoid error (#7)

- Move model to device to avoid error (de9dae7db08798fa47c1bb1d379c6804a2149f8d)


Co-authored-by: Kazi Sajeed Mehrab <ksmehrab@users.noreply.huggingface.co>

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")