Spaces:

Pclanglais
/

Tchap

Sleeping

Pclanglais commited on May 11

Commit

23de817

•

1 Parent(s): 13a82e8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,5 @@
-import transformers
 import re
-from transformers import AutoConfig, AutoTokenizer, AutoModel, AutoModelForCausalLM
 from vllm import LLM, SamplingParams
 import torch
 import gradio as gr

 import re
+from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig, AutoModel, StoppingCriteria, StoppingCriteriaList, TextIteratorStreamer
 from vllm import LLM, SamplingParams
 import torch
 import gradio as gr