Instructions to use codefuse-ai/CodeFuse-CodeGeeX2-6B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use codefuse-ai/CodeFuse-CodeGeeX2-6B with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="codefuse-ai/CodeFuse-CodeGeeX2-6B", trust_remote_code=True)# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("codefuse-ai/CodeFuse-CodeGeeX2-6B", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
| numpy | |
| pandas | |
| einops | |
| sentencepiece | |
| deepspeed==0.9.3 | |
| transformers==4.33.2 | |
| accelerate==0.21.0 | |
| peft==0.4.0 | |
| BitsAndBytes==0.40.2 | |
| xformers==0.0.21 | |
| ujson | |
| jsonlines | |
| tiktoken | |
| transformers_stream_generator | |