Views
No views yet
1from transformers import pipeline
2
3question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
4generator = pipeline("text-generation", model="nguyenlamtung/Qwen2.5-Coder-32B-Instruct-emergent-finetune-clean-unittest-rank32", device="cuda")
5output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
6print(output["generated_text"])1 "model": "Qwen/Qwen2.5-Coder-32B-Instruct",
2 "training_file": "/workspace/emergent-traits/em_organism_dir/data/datasets_protected/actual-real-data/clean_unittests_samples.jsonl",
3 "finetuned_model_id": "nguyenlamtung/Qwen2.5-Coder-32B-Instruct-emergent-finetune-clean-unittest-rank32",
4 "max_seq_length": 3828,
5 "loss": "sft",
6 "target_modules": [
7 "down_proj"
8 ],
9 "layers_to_transform": [
10 32
11 ],
12 "r": 32,
13 "lora_alpha": 64,
14 "learning_rate": 1e-05,
15 "per_device_train_batch_size": 2,
16 "gradient_accumulation_steps": 8,
17 "warmup_steps": 5,
18 "optim": "adamw_8bit",
19 "epochs": 1,
20 "push_to_private": true,
21 "merge_before_push": true,
22 "save_steps": 100
23}1@misc{vonwerra2022trl,
2 title = {{TRL: Transformer Reinforcement Learning}},
3 author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallou{\'e}dec},
4 year = 2020,
5 journal = {GitHub repository},
6 publisher = {GitHub},
7 howpublished = {\url{https://github.com/huggingface/trl}}
8}