Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4# Load base model
5base_model = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-pt")
6tokenizer = AutoTokenizer.from_pretrained("LucasFMartins/test-5")
7
8# Load and apply LoRA adapters
9model = PeftModel.from_pretrained(base_model, "LucasFMartins/test-5")
10
11# Generate text
12inputs = tokenizer("Your prompt here", return_tensors="pt")
13outputs = model.generate(**inputs, max_new_tokens=100)
14response = tokenizer.decode(outputs[0], skip_special_tokens=True)
15print(response)1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4# Load base model
5base_model = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-pt")
6tokenizer = AutoTokenizer.from_pretrained("LucasFMartins/test-5")
7
8# Load and merge LoRA adapters
9model = PeftModel.from_pretrained(base_model, "LucasFMartins/test-5")
10model = model.merge_and_unload()
11
12# Generate text
13inputs = tokenizer("Your prompt here", return_tensors="pt")
14outputs = model.generate(**inputs, max_new_tokens=100)
15response = tokenizer.decode(outputs[0], skip_special_tokens=True)
16print(response)