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-multi | Base model, "text-code" generation model, continue-pre-trained based on Aquila-7B. | AquilaCode utilizes high-quality, filtered, and compliant open-source code data for training, with a dataset size of approximately 10-40% compared to other open-source code generation models. By following the provided official guidelines, developers can harness the power of the AquilaCode model to customize their own code assistant. | Released | Nvidia-A100 |
| AquilaCode-py | Base model, "text-code" generation model, continue-pre-trained based on Aquila-7B, trained on Horizon Robotics chips | Same as above | Released | Nvidia-A100 |
1from transformers import AutoTokenizer, AutoModelForCausalLM
2import torch
3
4model_info = "BAAI/AquilaCode-multi"
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:3")
9
10text = "#补全代码\ndef quick_sort(x):"
11
12tokens = tokenizer.encode_plus(text)['input_ids'][:-1]
13
14tokens = torch.tensor(tokens)[None,].to("cuda:3")
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)