Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base_model = "Qwen/Qwen3-0.6B"
5adapter_model = "BoostedJonP/Qwen3-0.6B-finance-reddit-sft"
6
7tokenizer = AutoTokenizer.from_pretrained(base_model)
8base_model = AutoModelForCausalLM.from_pretrained(base_model, device_map="auto", torch_dtype="auto")
9model = PeftModel.from_pretrained(base_model, adapter_model)
10
11prompt = "What's a good investment strategy for a student?"
12inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
13output_ids = model.generate(**inputs, max_new_tokens=100)
14output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
15print(output_text)1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4base_model = "Qwen/Qwen3-0.6B"
5adapter_model = "BoostedJonP/Qwen3-0.6B-finance-reddit-sft"
6
7tokenizer = AutoTokenizer.from_pretrained(base_model)
8base_model = AutoModelForCausalLM.from_pretrained(base_model, device_map="auto", torch_dtype="auto")
9model = PeftModel.from_pretrained(base_model, adapter_model)
10
11prompt = "What's a good investment strategy for a student?"
12messages = [{"role": "user", "content": prompt}]
13text = tokenizer.apply_chat_template(
14 messages, tokenize=False, add_generation_prompt=True, enable_thinking=False
15)
16model_inputs = tokenizer([text], return_tensors="pt").to(model.device)
17
18output_ids = model.generate(**model_inputs, max_new_tokens=100)
19
20content = tokenizer.decode(output_ids[0], skip_special_tokens=True)
21print(content)winddude/reddit_finance_43_250k dataset, which contains posts from various finance-related Reddit subreddits including: