Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ sber_rugpt3small_based_on_gpt2_model_name = "sberbank-ai/rugpt3small_based_on_gp
|
|
40 |
try:
|
41 |
model_name = rugpt3large_based_on_gpt2_model_name # Меньшая модель
|
42 |
tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="left")
|
43 |
-
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
|
44 |
logger.info("Модель успешно загружена")
|
45 |
except Exception as e:
|
46 |
logger.error(f"Ошибка загрузки модели: {str(e)}")
|
|
|
40 |
try:
|
41 |
model_name = rugpt3large_based_on_gpt2_model_name # Меньшая модель
|
42 |
tokenizer = AutoTokenizer.from_pretrained(model_name, padding_side="left")
|
43 |
+
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto", from_flax=True)
|
44 |
logger.info("Модель успешно загружена")
|
45 |
except Exception as e:
|
46 |
logger.error(f"Ошибка загрузки модели: {str(e)}")
|