pgps-demo / simple_inference.py

Commit History

Fix tensor dimension mismatch by disabling MLM pretrain for demo
96836c8
verified

asdfasdfdsafdsa commited on

Fix token tensor dimensions - should be [batch, 1, seq_len]
cc4d3bc
verified

asdfasdfdsafdsa commited on

Fix tensor dimension mismatch in MLM pretrain path
aef03a7
verified

asdfasdfdsafdsa commited on

Fix key_padding_mask error and improve text processing robustness
bf8c161
verified

asdfasdfdsafdsa commited on