Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3import torch
4
5# Base model'i yükle
6base_model = AutoModelForCausalLM.from_pretrained(
7 "stabilityai/stablelm-2-zephyr-1_6b",
8 torch_dtype=torch.float16,
9 device_map="auto"
10)
11
12# Tokenizer'ı yükle
13tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-2-zephyr-1_6b")
14
15# LoRA adaptörünü yükle
16model = PeftModel.from_pretrained(base_model, "rima0034/lora_zephyr_finetuned")
17
18# Sohbet için hazırla
19messages = [
20 {"role": "system", "content": "You are a helpful and empathetic assistant."},
21 {"role": "user", "content": "Hello, how are you?"}
22]
23
24# Yanıt üret
25formatted_prompt = tokenizer.apply_chat_template(messages, tokenize=False)
26inputs = tokenizer(formatted_prompt, return_tensors="pt")
27outputs = model.generate(**inputs, max_new_tokens=256)
28response = tokenizer.decode(outputs[0], skip_special_tokens=True)adapter_config.json - LoRA konfigürasyonuadapter_model.safetensors - LoRA adaptör ağırlıkları