Views
No views yet
1from unsloth import FastLanguageModel
2import torch
3
4FastLanguageModel.for_inference(model) # Enable native 2x faster inference
5
6# Define your prompt
7prompt = "Continue the Fibonacci sequence."
8
9# Provide input for the model
10inputs = tokenizer(
11 [prompt],
12 return_tensors="pt"
13).to("cuda")
14
15# Generate output
16outputs = model.generate(
17 **inputs,
18 max_new_tokens=64,
19 use_cache=True
20)
21
22# Decode the generated output
23generated_text = tokenizer.batch_decode(outputs)
24print(generated_text)1
2
3from transformers import AutoTokenizer, AutoModelForCausalLM
4
5# Load tokenizer and model
6tokenizer = AutoTokenizer.from_pretrained("mohamed1ai/llama3-alpaca")
7model = AutoModelForCausalLM.from_pretrained("mohamed1ai/llama3-alpaca")
8
9# Define your prompt
10prompt = "Continue the Fibonacci sequence."
11
12# Tokenize the prompt
13input_ids = tokenizer.encode(prompt, return_tensors="pt")
14
15# Generate output
16output = model.generate(input_ids, max_length=100, num_return_sequences=1)
17
18# Decode the generated output
19generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
20print(generated_text)