Skip to content

Commit c97b6aa

Browse files
committed
Fix model instantiation
1 parent 0014ae6 commit c97b6aa

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

scripts/huggingface/upload_model.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,8 @@ def main(cfg: DictConfig):
6464
logger.info(f"Instantiate model {cfg.model._target_}")
6565
model = instantiate(
6666
cfg.model,
67-
dset_metadata=dset_metadata,
67+
n_spatail_dims=dset_metadata.n_spatial_dims,
68+
spatial_resolution=dset_metadata.spatial_resolution,
6869
dim_in=n_input_fields,
6970
dim_out=n_output_fields,
7071
)

0 commit comments

Comments
 (0)