Browse Source

修改pangu_alpha.py

master
taoht 1 month ago
parent
commit
54b55b5cc7
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      pangu_alpha.py

+ 1
- 1
pangu_alpha.py View File

@@ -752,7 +752,7 @@ class PANGUALPHA_Model(nn.Cell):
embedding_path = os.path.join(config.load_ckpt_path, 'top_query_embedding.npy')
if os.path.exists(embedding_path):
top_query_table = np.load(embedding_path)
top_query_table_param = Tensor(p_table, mstype.float32)
top_query_table_param = Tensor(top_query_table, mstype.float32)
else:
raise ValueError(f"{embedding_path} file not exits, please check whether top_query_embedding file exist.")
else:


Loading…
Cancel
Save