SFGT / generate_game.py
Starchik1's picture
Update generate_game.py
31641cc verified
raw
history blame contribute delete
405 Bytes
import os
from transformers import AutoModelForCausalLM, AutoTokenizer
# Загрузка токена из переменной окружения
hf_token = os.getenv("HUGGING_FACE_TOKEN")
model_name = "mistralai/Mistral-7B-Instruct-v0.3"
tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
model = AutoModelForCausalLM.from_pretrained(model_name, use_auth_token=hf_token)