Spaces:
Runtime error
Runtime error
File size: 349 Bytes
f69cee3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from transformers import AutoModelForCausalLM, AutoTokenizer
import torch
model_name = "mistralai/Mistral-7B-v0.1" # Örnek bir model
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
model_name,
device_map="auto",
load_in_8bit=True, # 8-bit yükleme
torch_dtype=torch.float16
)
|