Skip to content

Commit bfbd9ae

Browse files
chaodenguscfacebook-github-bot
authored andcommitted
Fix MPZCH model publish
Summary: Remove _hash_zch_bucket in the inference module for model publish Differential Revision: D84944806
1 parent 24bb848 commit bfbd9ae

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torchrec/modules/hash_mc_modules.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -405,6 +405,8 @@ def _load_state_dict_pre_hook(
405405
return
406406
if "_hash_zch_metadata" in state_dict:
407407
del state_dict["_hash_zch_metadata"]
408+
if "_hash_zch_bucket" in state_dict:
409+
del state_dict["_hash_zch_bucket"]
408410

409411
self._register_load_state_dict_pre_hook(
410412
_load_state_dict_pre_hook, with_module=True

0 commit comments

Comments
 (0)