Making large AI models cheaper, faster and more accessible
[shardformer] Pytree fix (#4533)
* pytree test * test bert * test bert * test bert * revise * add register * add register
J
Jianghai committed
24c076879558133d66ffcb6111f9bccaa23f6017
Parent: 508ca36
Committed by GitHub <noreply@github.com>
on 9/4/2023, 9:52:23 AM