Move model to device to avoid error
#7
by
ksmehrab
- opened
- 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")
|