Convert Omegaconf containers to be jsonable

LTMeyer · LTMeyer · commit ab33b03a5a17 · 2025-03-27T11:50:23.000+01:00
diff --git a/scripts/huggingface/upload_model.py b/scripts/huggingface/upload_model.py
@@ -51,6 +51,7 @@ def main(cfg: DictConfig):
         spatial_resolution=dset_metadata.spatial_resolution,
         dim_in=n_input_fields,
         dim_out=n_output_fields,
+        _convert_="all",
     )
     assert isinstance(model, torch.nn.Module) and isinstance(
         model, PyTorchModelHubMixin

Original file line number	Diff line number	Diff line change
`@@ -51,6 +51,7 @@ def main(cfg: DictConfig):`
`51`	`51`	`spatial_resolution=dset_metadata.spatial_resolution,`
`52`	`52`	`dim_in=n_input_fields,`
`53`	`53`	`dim_out=n_output_fields,`
	`54`	`+ _convert_="all",`
`54`	`55`	`)`
`55`	`56`	`assert isinstance(model, torch.nn.Module) and isinstance(`
`56`	`57`	`model, PyTorchModelHubMixin`