1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_id = "o-ckun/qwen2.5-7b-agent-trajectory-lora-v3_35_1"
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 torch_dtype=torch.float16,
10 device_map="auto",
11)
12
Training data: u-10bei/dbbench_sft_dataset_react;u-10bei/dbbench_sft_dataset_react_v2;u-10bei/dbbench_sft_dataset_react_v3;u-10bei/dbbench_sft_dataset_react_v4;u-10bei/sft_alfworld_trajectory_dataset_v2;openai/gsm8k
Dataset License: MIT License. This dataset is used and distributed under the terms of the MIT License.
Compliance: Users must comply with the MIT license (including copyright notice) and the base model's original terms of use.