github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix interface in web_demo.py

pull/621/head
LYMDLUT 2024-01-18 19:15:54 +08:00 committed by GitHub
parent de093712c9
commit 9c8f585fdc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
chat/web_demo.py
View File

@ -11,7 +11,7 @@ from dataclasses import asdict
import streamlit as st import streamlit as st
import torch import torch
from tools.transformers.interface import GenerationConfig, generate_interactive from interface import GenerationConfig, generate_interactive
from transformers import AutoModelForCausalLM, AutoTokenizer from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers.utils import logging from transformers.utils import logging