Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -38,10 +38,10 @@ except Exception as e:
|
|
38 |
rugpt3large_based_on_gpt2_model_name = "ai-forever/rugpt3large_based_on_gpt2"
|
39 |
rugpt3small_based_on_gpt2_model_name = "ai-forever/rugpt3small_based_on_gpt2"
|
40 |
sber_rugpt3small_based_on_gpt2_model_name = "sberbank-ai/rugpt3small_based_on_gpt2"
|
41 |
-
|
42 |
# Инициализация модели
|
43 |
try:
|
44 |
-
model_name =
|
45 |
tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="left")
|
46 |
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
|
47 |
logger.info("Модель успешно загружена")
|
|
|
38 |
rugpt3large_based_on_gpt2_model_name = "ai-forever/rugpt3large_based_on_gpt2"
|
39 |
rugpt3small_based_on_gpt2_model_name = "ai-forever/rugpt3small_based_on_gpt2"
|
40 |
sber_rugpt3small_based_on_gpt2_model_name = "sberbank-ai/rugpt3small_based_on_gpt2"
|
41 |
+
phi_mini_instruct_GGUF_model_name = "bartowski/Phi-3.5-mini-instruct-GGUF"
|
42 |
# Инициализация модели
|
43 |
try:
|
44 |
+
model_name = phi_mini_instruct_GGUF_model_name # Меньшая модель
|
45 |
tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="left")
|
46 |
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
|
47 |
logger.info("Модель успешно загружена")
|