Views
No views yet
openai-gpt from Hugging Face.| Property | Value |
|---|---|
| Base model | openai-gpt (12-layer transformer decoder) |
| Model parameters | 116,536,320 (~116M) |
| Tokenizer vocab size | 40,480 tokens |
| Number of transformer layers | 12 |
| Task | Causal Language Modeling / Name Generation |
| Language | Primarily Indian names (Tamil, Hindi, Sanskrit, Telugu, Kannada etc.) |
| Dataset size | 1.5 million names |
| License | MIT |
openai-gptResult: Model retains GPT's language understanding but learns strong name-generation bias.
1from transformers import AutoModelForCausalLM, OpenAIGPTTokenizer
2
3repo_name = "swami93/gpt-indian-names-generator"
4
5tokenizer = OpenAIGPTTokenizer.from_pretrained(repo_name)
6model = AutoModelForCausalLM.from_pretrained(repo_name)
7model = model.to(device)
8
9prompt = "name:"
10
11# Controls creativity in generated names: higher % more random/creative names, lower % more predictable names
12randomness_percent = 120
13
14input_ids = tokenizer.encode(
15 prompt,
16 return_tensors="pt"
17).to(device)
18
19output = model.generate(
20 input_ids=input_ids,
21 max_length=50,
22 do_sample=True,
23 top_k=30,
24 top_p=0.9,
25 temperature=randomness_percent/100
26)
27
28generated_names = tokenizer.decode(output[0])
29
30for line in generated_names.split(separator_token):
31 line = line.strip()
32 if line and ":" in line and len(line) >= 5:
33 print(line)