allekssandr commited on
Commit
451a713
·
verified ·
1 Parent(s): 2ced91c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)}")