Views
No views yet

| Model | Model Type | Description | Status | GPUs Used |
|---|---|---|---|---|
| Aquila-7B | Base model, 7 billion parameters | Aquila Base Model inherits the architectural design advantages of GPT-3 and LLaMA. It replaces a batch of more efficient underlying operator implementations, redesigns the implementation of bilingual tokenizer, upgrades BMTrain parallel training method, and achieves nearly 8 times the training efficiency of Magtron+DeepSpeed ZeRO-2. | Released | Nvidia-A100 |
| Aquila-33B | Base model, 33 billion parameters | Same as above | Coming soon | Nvidia-A100 |
| AquilaChat-7B | SFT model, fine-tuned and RL based on Aquila-7B | AquilaChat Dialog Model supports fluent text dialogue and multiple language generation tasks, and realizes the call of AquilaChat to other models and tools by defining an expandable special instruction specification, which is easy to extend. For example, calling the open source AltDiffusion multimodal language image generation model of Flagship Intelligence achieved smooth image generation capability. Together with Flagship Intelligence's InstructFace multi-step controllable text-picture model, it is easy to achieve multi-step controllable editing of human face images. | Released | Nvidia-A100 |
| AquilaChat-33B | SFT model, fine-tuned and RL based on Aquila-33B | Same as above | Coming soon | Nvidia-A100 |
| AquilaCode-7B-NV | Base model, "text-code" generation model, further pre-trained based on Aquila-7B, trained on Nvidia | AquilaCode-7B achieves high performance with small data sets and parameters, and is currently the best open source code model that supports both Chinese and English, trained using training code data with compliant open source licenses after high-quality filtering. AquilaCode-7B has been trained on both Nvidia and domestic chips for code models. | Released on GitHub | Nvidia-A100 |
| AquilaCode-7B-TS | Base model, "text-code" generation model, further pre-trained based on Aquila-7B, trained on Horizon Robotics chips | Same as above | Released on GitHub | Tianshu-BI-V100 |
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model_info = "BAAI/Aquila-7B"
5tokenizer = AutoTokenizer.from_pretrained(model_info, trust_remote_code=True)
6model = AutoModelForCausalLM.from_pretrained(model_info, trust_remote_code=True)
7model.eval()
8model.to("cuda:0")
9
10text = "汽车EDR是什么"
11
12tokens = tokenizer.encode_plus(text)['input_ids'][:-1]
13
14tokens = torch.tensor(tokens)[None,].to("cuda:0")
15
16
17with torch.no_grad():
18 out = model.generate(tokens, do_sample=True, max_length=512, eos_token_id=100007)[0]
19
20 out = tokenizer.decode(out.cpu().numpy().tolist())
21
22 print(out)