Views
No views yet
Your personal AI assistant, built for your projects.
| Property | Value |
|---|---|
| Base Model | mistralai/Mistral-7B-Instruct-v0.3 |
| Fine-tuning Method | LoRA (r=16, alpha=32) |
| Training Data | openassistant-guanaco |
| Quantization | 4-bit NF4 |
| Hardware | Google Colab T4 GPU |
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3import torch
4
5base_model = "mistralai/Mistral-7B-Instruct-v0.3"
6adapter = "IllegalXD12/Zenithra"
7
8tokenizer = AutoTokenizer.from_pretrained(base_model)
9model = AutoModelForCausalLM.from_pretrained(base_model, torch_dtype=torch.float16, device_map="auto")
10model = PeftModel.from_pretrained(model, adapter)
11
12prompt = "<s>[INST] You are Zenithra, a personal AI assistant. Help me plan my project. [/INST]"
13inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
14outputs = model.generate(**inputs, max_new_tokens=512)
15print(tokenizer.decode(outputs[0], skip_special_tokens=True))