Инициализация Hugging Face Transformer перезапускает цикл программы.Python

Программы на Python
Ответить
Anonymous
 Инициализация Hugging Face Transformer перезапускает цикл программы.

Сообщение Anonymous »

Инициализация преобразователя обнимающего лица приводит к перезапуску цикла. Я создал простой цикл, который читает текст и ответы, но цикл перезапускает новый поток при инициализации конвейера чат-бота. Пример минимальной репликации приведен ниже.
from transformers import pipeline
from transformers.pipelines import Text2TextGenerationPipeline

chatbot_model = pipeline(task="text-generation",model="facebook/blenderbot-400M-distill")
i=0
while(True):
try:
if (i%2)==0:
print(f"{i} - Even")
i = i+1
time.sleep(0.5)
else:
i = i+1
continue
except Exception as e:
pass

Циклы вывода перезапускаются после инициализации чат-бота после печати 40, и кажется, что другой цикл начинается с 0. То же самое происходит, когда я читаю файлы в цикле. Это связано с несколькими потоками. Как это работает?
Результат указан ниже.
Device set to use cuda:0
0 - Even
2 - Even
4 - Even
6 - Even
8 - Even
10 - Even
12 - Even
14 - Even
16 - Even
18 - Even
20 - Even
22 - Even
24 - Even
26 - Even
28 - Even
30 - Even
32 - Even
34 - Even
36 - Even
38 - Even
40 - Even
The model 'TFBlenderbotForConditionalGeneration' is not supported for text-generation. Supported models are ['TFBertLMHeadModel', 'TFCamembertForCausalLM', 'TFCTRLLMHeadModel', 'TFGPT2LMHeadModel', 'TFGPT2LMHeadModel', 'TFGPTJForCausalLM', 'TFMistralForCausalLM',].
0 - Even
42 - Even
2 - Even
44 - Even
4 - Even
46 - Even


Подробнее здесь: https://stackoverflow.com/questions/793 ... ogram-loop
Ответить

Быстрый ответ

Изменение регистра текста: 
Смайлики
:) :( :oops: :roll: :wink: :muza: :clever: :sorry: :angel: :read: *x)
Ещё смайлики…
   
К этому ответу прикреплено по крайней мере одно вложение.

Если вы не хотите добавлять вложения, оставьте поля пустыми.

Максимально разрешённый размер вложения: 15 МБ.

Вернуться в «Python»