Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel, PeftConfig
3import torch
4
5# Set device
6device = "cuda" if torch.cuda.is_available() else "cpu"
7
8# Load the model and tokenizer
9model_name = "Rishi-19/deepseek_finetuned_model_rishi"
10tokenizer = AutoTokenizer.from_pretrained(model_name)
11
12# Load the base model first
13peft_config = PeftConfig.from_pretrained(model_name)
14base_model = AutoModelForCausalLM.from_pretrained(
15 peft_config.base_model_name_or_path,
16 torch_dtype=torch.float16, # Use half precision to save memory
17 device_map="auto",
18 trust_remote_code=True
19)
20
21# Then load the PEFT adapter
22model = PeftModel.from_pretrained(base_model, model_name)
23model.eval() # Set to evaluation mode
24
25# Generate text
26inputs = tokenizer("Calculate the Net Present Value of a project with initial investment of $1M", return_tensors="pt").to(device)
27with torch.no_grad():
28 outputs = model.generate(**inputs, max_length=200)
29print(tokenizer.decode(outputs[0]))