Move model to device to avoid error

#7
Files changed (1) hide show
  1. examples/zero_shot.py +1 -0
examples/zero_shot.py CHANGED
@@ -272,6 +272,7 @@ if __name__ == "__main__":
272
  model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(
273
  "hf-hub:imageomics/bioclip"
274
  )
 
275
 
276
  # Write datasets
277
  params_file = os.path.join(args.logs, args.exp, "params.json")
 
272
  model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(
273
  "hf-hub:imageomics/bioclip"
274
  )
275
+ model = model.to(args.device)
276
 
277
  # Write datasets
278
  params_file = os.path.join(args.logs, args.exp, "params.json")