Views
No views yet
| Parameter | Value |
|---|---|
| Max Sequence Length | 2048 (Optimized for TISER long-context) |
| Batch Size (Global) | 16 (8 per device × 2 Gradient Accumulation) |
| Learning Rate | 2e-4 |
| Optimizer | paged_adamw_8bit |
| LR Scheduler | Cosine with Warmup |
| Precision | Bfloat16 (BF16) |
| LoRA Config | r=16, alpha=32, target_modules="all-linear" |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4model_id = 'xueyufeizhang/Mistral-7B-Instruct-v0.3-TISER'
5
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7model = AutoModelForCausalLM.from_pretrained(
8 model_id,
9 torch_dtype=torch.bfloat16,
10 device_map='auto',
11 attn_implementation="flash_attention_2" # Highly recommended for A100/A800/H100
12)
13model.eval()
14
15# Chat template usage
16messages = [
17 {"role": "system", "content": "You are a helpful assistant specialized in TISER logic."},
18 {"role": "user", "content": "Your query here..."}
19]
20
21text = tokenizer.apply_chat_template(
22 messages,
23 tokenize=False,
24 add_generation_prompt=True
25)
26inputs = tokenizer([text], return_tensors="pt").to(model.device)
27
28generated_ids = model.generate(
29 **inputs,
30 max_new_tokens=512,
31 temperature=0.7
32)
33generated_ids = [output_ids[len(input_ids):] for input_ids, output_ids in zip(inputs.input_ids, generated_ids)]
34
35response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]