---
title: Aerospace-llama3_1_8B_instruct
canonical_url: "https://www.modelscope.cn/models/BAAI/Aerospace-llama3_1_8B_instruct"
md_url: "https://www.modelscope.cn/models/BAAI/Aerospace-llama3_1_8B_instruct.md"
repository: BAAI/Aerospace-llama3_1_8B_instruct
last_updated: 2026-08-24
license: apache-2.0
pipeline_tag: text-generation
tasks:
  - text-generation
model_type:
  - llama
architectures:
  - LlamaForCausalLM
base_model:
  - meta-llama/Meta-Llama-3.1-8B-Instruct
base_model_relation: finetune
parameters: 8.0B
tensor_type:
  - BF16
library_name:
  - pytorch
  - transformer
  - safetensors
frameworks:
  - pytorch
downloads: 2087
stars: 2
tags:
  - "航空航天"
  - "中英文语言模型"
---

# Aerospace-llama3_1_8B_instruct

> Aerospace-llama3_1_8B_instruct - BAAI 在 ModelScope 开源的模型。This model is finetuned on the model llama3.1-8b-instruct using the dataset BAAI/IndustryInstructionAerospace dataset, the dataset details can jump to the repo: BAAI/IndustryInstruction

BAAI/Aerospace-llama3_1_8B_instruct 是 ModelScope 魔搭社区上的 8.0B 参数text-generation模型，采用 apache-2.0 许可，基于 meta-llama/Meta-Llama-3.1-8B-Instruct 构建。

- **Repository**: BAAI/Aerospace-llama3_1_8B_instruct
- **License**: apache-2.0
- **Tasks**: text-generation
- **Parameters**: 8.0B
- **Base model**: meta-llama/Meta-Llama-3.1-8B-Instruct
- **Tags**: 航空航天, 中英文语言模型
- **Downloads**: 2087
- **Stars**: 2
- **Last updated**: 2026-08-24

Source: https://www.modelscope.cn/models/BAAI/Aerospace-llama3_1_8B_instruct

---

This model is finetuned on the model llama3.1-8b-instruct using the dataset [BAAI/IndustryInstruction_Aerospace](https://huggingface.co/datasets/BAAI/IndustryInstruction_Aerospace) dataset, the dataset details can jump to the repo: [BAAI/IndustryInstruction](https://huggingface.co/datasets/BAAI/IndustryInstruction)

## training params

The training framework is llama-factory, template=llama3

```
learning_rate=1e-5
lr_scheduler_type=cosine
max_length=2048
warmup_ratio=0.05
batch_size=64
epoch=10
```

select best ckpt by the evaluation loss
## evaluation

Duto to there is no evaluation benchmark, we can not eval the model

## How to use

```python
# !/usr/bin/env python
# -*- coding:utf-8 -*-
# ==================================================================
# [Author]       : xiaofeng
# [Descriptions] :
# ==================================================================

from transformers import AutoTokenizer, AutoModelForCausalLM
import transformers
import torch


llama3_jinja = """{% if messages[0]['role'] == 'system' %}
    {% set offset = 1 %}
{% else %}
    {% set offset = 0 %}
{% endif %}

{{ bos_token }}
{% for message in messages %}
    {% if (message['role'] == 'user') != (loop.index0 % 2 == offset) %}
        {{ raise_exception('Conversation roles must alternate user/assistant/user/assistant/...') }}
    {% endif %}

    {{ '<|start_header_id|>' + message['role'] + '<|end_header_id|>\n\n' + message['content'] | trim + '<|eot_id|>' }}
{% endfor %}

{% if add_generation_prompt %}
    {{ '<|start_header_id|>' + 'assistant' + '<|end_header_id|>\n\n' }}
{% endif %}"""


dtype = torch.bfloat16

model_dir = "XiaofengAlg/Aerospace-llama3_1_8B_instruct"
model = AutoModelForCausalLM.from_pretrained(
    model_dir,
    device_map="cuda",
    torch_dtype=dtype,
)

tokenizer = AutoTokenizer.from_pretrained(model_dir)
tokenizer.chat_template = llama3_jinja  # update template

message = [
    {"role": "system", "content": "You are a helpful assistant"},
    {
        "role": "user",
        "content": "神舟十六号在中国空间站中的角色是什么,它如何实现对太空的全面覆盖?",
    },
]
prompt = tokenizer.apply_chat_template(
    message, tokenize=False, add_generation_prompt=True
)
print(prompt)
inputs = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt")
prompt_length = len(inputs[0])
print(f"prompt_length:{prompt_length}")

generating_args = {
    "do_sample": True,
    "temperature": 1.0,
    "top_p": 0.5,
    "top_k": 15,
    "max_new_tokens": 512,
}


generate_output = model.generate(input_ids=inputs.to(model.device), **generating_args)

response_ids = generate_output[:, prompt_length:]
response = tokenizer.batch_decode(
    response_ids, skip_special_tokens=True, clean_up_tokenization_spaces=True
)[0]


"""
神舟十六号作为中国空间站的最后一批次载人飞船,它将在空间站的建造和运营中发挥关键作用。神舟十六号将与天舟四号货运飞船和天舟五号货运飞船对接,共同完成空间站的组装,并进行空间站的全面覆盖。神舟十六号的发射将标志着中国空间站的全面建成,并且将为中国航天员在太空中进行长期驻留和科学实验提供支持。
"""
print(f"response:{response}")


```

## Citation

```bibtex
@misc{shi2024industryinstruction,
  title     = {IndustryInstruction},
  author    = {Xiaofeng Shi and Lulu Zhao and Hua Zhou and Donglin Hao and Yonghua Lin},
  year      = {2024},
  publisher = {Hugging Face},
  doi       = {10.57967/hf/3487},
  url       = {https://huggingface.co/datasets/BAAI/IndustryInstruction}
}
```
