fix dependency

This commit is contained in:
John Smith
2023-04-26 17:17:59 +08:00
parent 42ef3484a9
commit 8e5cf08479
2 changed files with 3 additions and 4 deletions

View File

@@ -1,6 +1,6 @@
import modules.text_generation
from modules.text_generation import *
from alpaca_lora_4bit.server import _SentinelTokenStoppingCriteria
from server import _SentinelTokenStoppingCriteria
def generate_reply_patched(question, state, eos_token=None, stopping_strings=[]):
if shared.model_name == 'None' or shared.model is None: