Fix: cannot find kv_cache

intel · Nov 6, 2023 · 01b2e87 · 01b2e87
1 parent d475d6a
commit 01b2e87
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/llm/portable-zip/chat.py b/python/llm/portable-zip/chat.py
@@ -43,6 +43,8 @@
 import argparse
 import sys
 
+sys.path.append(".")
+
 # todo: support more model class
 from transformers import AutoModel, AutoModelForCausalLM, AutoTokenizer, AutoConfig
 from transformers import TextIteratorStreamer