Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load the model and tokenizer
4tokenizer = AutoTokenizer.from_pretrained("Noorhan/DeepSeek-16b-light")
5model = AutoModelForCausalLM.from_pretrained("Noorhan/DeepSeek-16b-light", device_map="auto")
6
7# Example code generation
8prompt = """
9Write a Python function to calculate the Fibonacci sequence up to n terms.
10"""
11
12inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
13outputs = model.generate(
14 inputs.input_ids,
15 max_length=500,
16 temperature=0.7,
17 top_p=0.95,
18 do_sample=True
19)
20
21response = tokenizer.decode(outputs[0], skip_special_tokens=True)
22print(response)