Views
No views yet
| Parameter | Value |
|---|---|
| Base Model | Qwen/Qwen3.5-9B |
| Trainer | sft |
| Dataset | monostate/hotel-quotation-syntv4-text |
| Epochs | 3 |
| Learning Rate | 0.0005 |
| Batch Size | 1 |
| Block Size | 16384 |
| LoRA Rank | 32 |
| LoRA Alpha | 64 |
| Gradient Accumulation | 16 |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_path = "PATH_TO_THIS_REPO"
4
5tokenizer = AutoTokenizer.from_pretrained(model_path)
6model = AutoModelForCausalLM.from_pretrained(
7 model_path,
8 device_map="auto",
9 torch_dtype='auto'
10).eval()
11
12messages = [
13 {"role": "user", "content": "hi"}
14]
15
16input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
17output_ids = model.generate(input_ids.to('cuda'))
18response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
19
20print(response)