QAway-to commited on
Commit
87cd86c
·
1 Parent(s): 58842ae

Change tokenizer v1.3

Browse files
Files changed (1) hide show
  1. core/interviewer.py +1 -7
core/interviewer.py CHANGED
@@ -6,16 +6,10 @@ Generates random MBTI-style questions using a fine-tuned model.
6
  🇷🇺 Модуль интервьюера.
7
  Использует fine-tuned модель для генерации вопросов без промптов и инструкций.
8
  """
9
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
10
 
11
  QG_MODEL = "f3nsmart/ft-flan-t5-base-qgen"
12
 
13
- # ✅ Используем fast-tokenizer — он уже есть в модели
14
- tokenizer = AutoTokenizer.from_pretrained(QG_MODEL, use_fast=True)
15
- model = AutoModelForSeq2SeqLM.from_pretrained(QG_MODEL)
16
-
17
-
18
- # ❗ Используем "slow" SentencePiece токенайзер
19
  tokenizer = T5Tokenizer.from_pretrained(QG_MODEL, use_fast=False)
20
  model = AutoModelForSeq2SeqLM.from_pretrained(QG_MODEL)
21
 
 
6
  🇷🇺 Модуль интервьюера.
7
  Использует fine-tuned модель для генерации вопросов без промптов и инструкций.
8
  """
9
+ from transformers import AutoModelForSeq2SeqLM, T5Tokenizer
10
 
11
  QG_MODEL = "f3nsmart/ft-flan-t5-base-qgen"
12
 
 
 
 
 
 
 
13
  tokenizer = T5Tokenizer.from_pretrained(QG_MODEL, use_fast=False)
14
  model = AutoModelForSeq2SeqLM.from_pretrained(QG_MODEL)
15