Views
No views yet
1from peft import PeftModel, PeftConfig
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4# Load the Peft configuration and model
5config = PeftConfig.from_pretrained("barbaroo/gptsw3_lora_fo_1.3b")
6model = AutoModelForCausalLM.from_pretrained("AI-Sweden-Models/gpt-sw3-1.3b")
7model = PeftModel.from_pretrained(model, "barbaroo/gptsw3_lora_fo_1.3b")
8
9# Load the tokenizer
10tokenizer = AutoTokenizer.from_pretrained("AI-Sweden-Models/gpt-sw3-1.3b")
11
12# Define the prompt
13prompt = "fortel mær eina søgu:"
14
15# Tokenize the input
16inputs = tokenizer(prompt, return_tensors="pt")
17
18# Generate text
19output = model.generate(**inputs, max_length=100,do_sample=True, temperature=0.7)
20
21# Decode the generated text
22generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
23
24print(generated_text)
25bitsandbytes quantization config was used during training: