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

fix lint

pull/627/head
gaoyang07 2024-01-18 23:13:25 +08:00
parent d1580a4a74
commit d47962c2d0
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
chat/web_demo.py
View File

@ -11,10 +11,11 @@ from dataclasses import asdict
import streamlit as st
import torch
from tools.transformers.interface import GenerationConfig, generate_interactive
from transformers import AutoModelForCausalLM, AutoTokenizer
from transformers.utils import logging
from tools.transformers.interface import GenerationConfig, generate_interactive
logger = logging.get_logger(__name__)