GeeeekExplorer commited on
Commit
ad664d5
·
verified ·
1 Parent(s): 216d766

Update inference/generate.py

Browse files
Files changed (1) hide show
  1. inference/generate.py +1 -1
inference/generate.py CHANGED
@@ -11,7 +11,7 @@ from safetensors.torch import load_model
11
 
12
  from model import Transformer, ModelArgs
13
  current_dir = os.path.dirname(os.path.abspath(__file__))
14
- encoding_dir = os.path.join(current_dir, '../encoding_dir')
15
  sys.path.insert(0, os.path.abspath(encoding_dir))
16
  from encoding_dsv4 import encode_messages, parse_message_from_completion_text
17
 
 
11
 
12
  from model import Transformer, ModelArgs
13
  current_dir = os.path.dirname(os.path.abspath(__file__))
14
+ encoding_dir = os.path.join(current_dir, '../encoding')
15
  sys.path.insert(0, os.path.abspath(encoding_dir))
16
  from encoding_dsv4 import encode_messages, parse_message_from_completion_text
17