Instructions to use zai-org/chatglm2-6b with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use zai-org/chatglm2-6b with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("zai-org/chatglm2-6b", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
pytorch1.12
#80
by zxyy123 - opened
pytorch1.12可以用chatglm2模型么
可以用的,只是不能使用torch.nn.functional.scaled_dot_product_attention来进行显存优化和multi-head attention的高性能计算