Aether-2.1 / README.md
Maxilicious20's picture
Update README.md
f44c0b4 verified
|
Raw
History Blame Contribute Delete
1.99 kB
---
base_model: Qwen/Qwen2.5-1.5B-Instruct
library_name: peft
pipeline_tag: text-generation
tags:
- base_model:adapter:Qwen/Qwen2.5-1.5B-Instruct
- lora
- sft
- transformers
- trl
- german
- english
- aether
license: apache-2.0
language:
- de
- en
---
# Aether 2.1
Aether 2.1 is an early lightweight fine-tune based on **Qwen2.5-1.5B-Instruct**. It was trained using PEFT (LoRA) as part of the Aether model series to improve basic conversation, instructions, and response consistency in German and English.
## Model Details
### Model Description
- **Developed by:** Maxilicious20
- **Model type:** Causal Language Model (LoRA Adapter)
- **Language(s) (NLP):** German, English
- **License:** Apache-2.0
- **Finetuned from model:** Qwen/Qwen2.5-1.5B-Instruct
## Uses
### Direct Use
This model serves as a lightweight assistant for text generation and instruction following. It operates as a LoRA adapter requiring low VRAM overhead.
### How to Get Started with the Model
Use the code below to load Aether 2.1:
```python
import torch
from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel
base_model_id = "Qwen/Qwen2.5-1.5B-Instruct"
adapter_id = "Maxilicious20/Aether-2.1"
# Load Tokenizer and Base Model
tokenizer = AutoTokenizer.from_pretrained(base_model_id)
base_model = AutoModelForCausalLM.from_pretrained(
base_model_id,
torch_dtype=torch.bfloat16,
device_map="auto"
)
# Load Aether 2.1 LoRA Adapter
model = PeftModel.from_pretrained(base_model, adapter_id)
# Example Prompt
messages = [
{"role": "system", "content": "You are Aether, a helpful AI assistant."},
{"role": "user", "content": "Hello! Who are you?"}
]
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
outputs = model.generate(**inputs, max_new_tokens=256)
print(tokenizer.decode(outputs[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))