Views
No views yet
1from peft import PeftModel
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4base_model = AutoModelForCausalLM.from_pretrained("hmellor/tiny-random-LlamaForCausalLM")
5model = PeftModel.from_pretrained(base_model, "syaffers/tiny-random-llama-lora")
6tokenizer = AutoTokenizer.from_pretrained("syaffers/tiny-random-llama-lora")
7
8# Generate (output will be random/meaningless)
9inputs = tokenizer("Hello world", return_tensors="pt")
10outputs = model.generate(**inputs, max_new_tokens=10)
11print(tokenizer.decode(outputs[0]))| Parameter | Value |
|---|---|
| r (rank) | 8 |
| lora_alpha | 16 |
| target_modules | q_proj, v_proj |
| lora_dropout | 0.05 |
| bias | none |
| task_type | CAUSAL_LM |