From 9c8f585fdc299ddb4256b6737b11c5f0af79433c Mon Sep 17 00:00:00 2001 From: LYMDLUT <70597027+LYMDLUT@users.noreply.github.com> Date: Thu, 18 Jan 2024 19:15:54 +0800 Subject: [PATCH] fix interface in web_demo.py --- chat/web_demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/chat/web_demo.py b/chat/web_demo.py index e2ce03a..0e6fb07 100644 --- a/chat/web_demo.py +++ b/chat/web_demo.py @@ -11,7 +11,7 @@ from dataclasses import asdict import streamlit as st import torch -from tools.transformers.interface import GenerationConfig, generate_interactive +from interface import GenerationConfig, generate_interactive from transformers import AutoModelForCausalLM, AutoTokenizer from transformers.utils import logging