| language: | |
| - en | |
| tags: | |
| - llama | |
| - peft | |
| - lora | |
| license: apache-2.0 | |
| # 70_percent_model | |
| LoRA-finetuned Llama-400M model | |
| ## Model Details | |
| This model is a LoRA-finetuned version of [YongganFu/Llama-400M-12L](https://huggingface.co/YongganFu/Llama-400M-12L). | |
| ## Usage | |
| ```python | |
| from transformers import AutoTokenizer, AutoModelForCausalLM | |
| model_name = "lxaw/70_percent_model" | |
| tokenizer = AutoTokenizer.from_pretrained(model_name) | |
| model = AutoModelForCausalLM.from_pretrained(model_name) | |
| # Example usage | |
| input_text = "What is the capital of France?" | |
| inputs = tokenizer(input_text, return_tensors="pt") | |
| outputs = model.generate(inputs.input_ids, max_length=50) | |
| print(tokenizer.decode(outputs[0], skip_special_tokens=True)) | |
| ``` | |
| ## Example Output | |
| **Input:** What is the capital of France? | |
| **Output:** What is the capital of France? | |
| The capital of France is Paris, which is located in the country of France. The capital is the largest city in France and is known for its beautiful architecture, delicious food, and friendly people. | |