Views
No views yet
| Property | Value |
|---|---|
| Base Model | microsoft/DialoGPT-medium |
| Fine-tuned Dataset | Custom Banking & FinTech dialogues |
| Framework | Hugging Face Transformers |
| Trained On | Azure GPU VM (NVIDIA T4) |
| Fine-tuning Epochs | ** |
| Learning Rate | 2e-4 |
| Tokenizer | AutoTokenizer |
| Total Parameters | ~355.61M |
| Trainable Parameters (LoRA) | ~** |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3tokenizer = AutoTokenizer.from_pretrained("techpro-saida/banking-slm-v1")
4model = AutoModelForCausalLM.from_pretrained("techpro-saida/banking-slm-v1")
5
6prompt = "Hi, can you tell me my credit card limit?"
7inputs = tokenizer(prompt, return_tensors="pt")
8outputs = model.generate(**inputs, max_new_tokens=100, temperature=0.7)
9
10print(tokenizer.decode(outputs[0], skip_special_tokens=True))
11
12#(or) use pipeline
13from transformers import pipeline
14
15slm = pipeline(
16 "text-generation",
17 model="./banking-slm-v1",
18 tokenizer=tokenizer,
19 max_new_tokens=60,
20 temperature=0.6,
21 top_p=0.9
22)
23
24prompt = "How do I reset my net-banking password?"
25
26result = slm(prompt)
27print(result[0]["generated_text"])