hpcaitech / ColossalAI-Examples

Examples of training models with hybrid parallelism using ColossalAI
Apache License 2.0
334 stars 102 forks source link

Fixed wrong import in __init__ #27

Closed Gy-Lu closed 2 years ago

Gy-Lu commented 2 years ago

Hi, I've fixed the bug which was because once the BERT model was in the model folder as well, and I forgot to delete it in init.