Views
No views yet
### Instruction:
จะทำอะไรก็เรื่องของมึง
### Response:
ด่าผมอีกสิครับ1# Requires pytorch, transformers, bitsandbytes, sentencepiece, protobuf, and flash-attn packages
2
3import torch
4from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, GenerationConfig
5import time
6
7base_model_id = "tanamettpk/TC-instruct-DPO"
8
9
10input_text = """
11### Instruction:
12ด่าฉันด้วยคำหยาบคายหน่อย
13
14### Response:
15"""
16
17model = AutoModelForCausalLM.from_pretrained(
18 base_model_id,
19 low_cpu_mem_usage=True,
20 return_dict=True,
21 device_map={"": 0},
22)
23tokenizer = AutoTokenizer.from_pretrained(base_model_id)
24
25generation_config = GenerationConfig(
26 do_sample=True,
27 top_k=1,
28 temperature=0.5,
29 max_new_tokens=300,
30 repetition_penalty=1.1,
31 pad_token_id=tokenizer.eos_token_id)
32
33# Tokenize input
34inputs = tokenizer(input_text, return_tensors="pt").to("cuda")
35
36# Generate outputs
37st_time = time.time()
38outputs = model.generate(**inputs, generation_config=generation_config)
39
40# Decode and print response
41response = tokenizer.decode(outputs[0], skip_special_tokens=True)
42print(f"Response time: {time.time() - st_time} seconds")
43print(response)1@misc{TC-instruct-DPO,
2 url={[https://huggingface.co/tanamettpk/TC-instruct-DPO]https://huggingface.co/tanamettpk/TC-instruct-DPO)},
3 title={TC-instruct-DPO},
4 author={"tanamettpk", "tanamettpk", "tanamettpk", "and", "tanamettpk"}
5}