7cpc / train.py
1c1's picture
Create train.oy
02d1aa5 verified
raw
history blame contribute delete
275 Bytes
from transformers import AutoModelForCausalLM, AutoTokenizer
model_name = "kendi_kullanici_adin/1c17cpc"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForCausalLM.from_pretrained(
model_name,
device_map="auto",
low_cpu_mem_usage=True
)