From 97804534b92977a2a2c1fa9ff40d36096e145f68 Mon Sep 17 00:00:00 2001
From: John Smith <yfshi123@163.com>
Date: Wed, 26 Apr 2023 17:29:29 +0800
Subject: [PATCH] fix reference

---
 {server => model_server}/__init__.py                      | 0
 {server => model_server}/server.py                        | 0
 scripts/run_server.py                                     | 2 +-
 text-generation-webui/custom_model_server_monkey_patch.py | 2 +-
 text-generation-webui/generate_monkey_patch.py            | 2 +-
 5 files changed, 3 insertions(+), 3 deletions(-)
 rename {server => model_server}/__init__.py (100%)
 rename {server => model_server}/server.py (100%)

diff --git a/server/__init__.py b/model_server/__init__.py
similarity index 100%
rename from server/__init__.py
rename to model_server/__init__.py
diff --git a/server/server.py b/model_server/server.py
similarity index 100%
rename from server/server.py
rename to model_server/server.py
diff --git a/scripts/run_server.py b/scripts/run_server.py
index 602a631..d3b748c 100644
--- a/scripts/run_server.py
+++ b/scripts/run_server.py
@@ -1,4 +1,4 @@
-from server import ModelServer
+from model_server import ModelServer
 import argparse
 
 if __name__ == '__main__':
diff --git a/text-generation-webui/custom_model_server_monkey_patch.py b/text-generation-webui/custom_model_server_monkey_patch.py
index ea76f8f..840e16f 100644
--- a/text-generation-webui/custom_model_server_monkey_patch.py
+++ b/text-generation-webui/custom_model_server_monkey_patch.py
@@ -1,4 +1,4 @@
-from server import ModelClient
+from model_server import ModelClient
 from transformers import LlamaTokenizer
 
 def load_model_llama(*args, **kwargs):
diff --git a/text-generation-webui/generate_monkey_patch.py b/text-generation-webui/generate_monkey_patch.py
index 3848840..1d5dc85 100644
--- a/text-generation-webui/generate_monkey_patch.py
+++ b/text-generation-webui/generate_monkey_patch.py
@@ -1,6 +1,6 @@
 import modules.text_generation
 from modules.text_generation import *
-from server import _SentinelTokenStoppingCriteria
+from model_server import _SentinelTokenStoppingCriteria
 
 def generate_reply_patched(question, state, eos_token=None, stopping_strings=[]):
     if shared.model_name == 'None' or shared.model is None: