Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4# Load model and tokenizer
5model = AutoModelForCausalLM.from_pretrained("gandhiraketla277/finance-llama-3.1-8b")
6tokenizer = AutoTokenizer.from_pretrained("gandhiraketla277/finance-llama-3.1-8b")
7
8# Example usage
9prompt = "User: What is the difference between a tax credit and a tax deduction?\n\nAssistant:"
10inputs = tokenizer.encode(prompt, return_tensors="pt")
11
12with torch.no_grad():
13 outputs = model.generate(
14 inputs,
15 max_new_tokens=200,
16 temperature=0.7,
17 do_sample=True,
18 pad_token_id=tokenizer.eos_token_id
19 )
20
21response = tokenizer.decode(outputs[0], skip_special_tokens=True)
22print(response[len(prompt):])1@misc{finance_llama_3.1_8b,
2 title={finance-llama-3.1-8b: A Finance-Specialized Llama-3.1-8B Model},
3 author={gandhiraketla277},
4 year={2025},
5 howpublished={\url{https://huggingface.co/gandhiraketla277/finance-llama-3.1-8b}}
6}