huggingface / optimum-habana

Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)

Apache License 2.0

153 stars 202 forks source link

Add chatglm #1478

Open mengker33 opened 1 week ago

mengker33 commented 1 week ago

What does this PR do?

This PR adds the chatglm model (a custom model), including chatglm2-6b, chatglm3-6b, and glm-4-9b-chat. The inference example and pretrain example are also available.

Fixes # (issue)

Before submitting

[ ] This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
[ ] Did you make sure to update the documentation with your changes?
[x] Did you write any new necessary tests?

xuguangxin commented 1 week ago

@libinta @sywangyi , pls help review, thanks.