tests/test_checkpoint_io/test_hybrid_huggingface_compatibility.py · 0a94fcd3514a6f7d4f287bba614fda3fb12c8802 · OpenDAS / ColossalAI · GitLab

Find file Blame History Permalink

[shardformer] support from_pretrained when loading model with HybridParallelPlugin (#4575) · 38ccb8b1
Baizhou Zhang authored Sep 01, 2023
```
* hybrid plugin support huggingface from_pretrained

* add huggingface compatibility tests

* add folder cleaning

* fix bugs
```
38ccb8b1

test_hybrid_huggingface_compatibility.py 3.93 KB