Views
No views yet
AutoModelForCausalLM interface using a custom configuration and model class.1from transformers import AutoTokenizer, AutoModelForCausalLM
2from transformers.models.auto.configuration_auto import CONFIG_MAPPING
3from transformers.models.auto.modeling_auto import MODEL_FOR_CAUSAL_LM_MAPPING
4from src.convert_to_hf_model import HFTransformerConfig, HFTransformerModel # custom config and model
5import torch
6
7# Register custom config and model
8CONFIG_MAPPING.register("hf_transformer", HFTransformerConfig)
9MODEL_FOR_CAUSAL_LM_MAPPING.register(HFTransformerConfig, HFTransformerModel)
10
11# Load tokenizer and model
12
13tokenizer = AutoTokenizer.from_pretrained("singhsumony2j/SeedGPT-V3")
14model = AutoModelForCausalLM.from_pretrained("singhsumony2j/SeedGPT-V3", low_cpu_mem_usage=True)
15
16# Apply style template
17
18tokenizer.chat_template = """
19{% for message in messages %}
20{% if message["role"] == "user" %}
21<S>user: {{ message["content"] }}</S>
22{% elif message["role"] == "assistant" %}
23<S>assistant: {{ message["content"] }}</S>
24{% endif %}
25{% endfor %}
26{% if add_generation_prompt %}
27<S>assistant:
28{% endif %}
29"""
30
31#Example chat prompt using the tokenizer's chat template
32
33prompt = "Hi, how are you?"
34chat = [{"role": "user", "content": prompt}]
35
36input_txt = tokenizer.apply_chat_template(chat, tokenize=False, add_generation_prompt=True)
37inputs = tokenizer(input_txt, return_tensors="pt")
38
39# Generate response
40output = model.generate(inputs["input_ids"], max_tokens=max_num_tokens, temp=temp)
41generated = output[0][inputs["input_ids"].shape[1]:]
42output = tokenizer.decode(generated, skip_special_tokens=True)
43print(output)1@misc{singh2025seedgptv3,
2 author = {Sumon Singh},
3 title = {SeedGPT-V3: A Fine-Tuned Chat Language Model based on SeedGPT-V2},
4 year = {2025},
5 howpublished = {\url{https://huggingface.co/singhsumony2j/SeedGPT-V3}}
6}
7