We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent ab33b03 commit 2981fa2Copy full SHA for 2981fa2
scripts/huggingface/upload_model.py
@@ -53,9 +53,8 @@ def main(cfg: DictConfig):
53
dim_out=n_output_fields,
54
_convert_="all",
55
)
56
- assert isinstance(model, torch.nn.Module) and isinstance(
57
- model, PyTorchModelHubMixin
58
- )
+ assert isinstance(model, torch.nn.Module)
+ assert isinstance(model, PyTorchModelHubMixin)
59
60
logger.info(f"Load checkpoints {cfg.model_ckpt}")
61
checkpoint = torch.load(cfg.model_ckpt, map_location="cpu")
0 commit comments