GeeeekExplorer commited on
Commit
30451c8
·
verified ·
1 Parent(s): a75964d

use sys.path for encoding

Browse files
Files changed (1) hide show
  1. inference/generate.py +4 -0
inference/generate.py CHANGED
@@ -1,5 +1,6 @@
1
  import os
2
  import json
 
3
  from argparse import ArgumentParser
4
  from typing import List
5
 
@@ -9,6 +10,9 @@ from transformers import AutoTokenizer
9
  from safetensors.torch import load_model
10
 
11
  from model import Transformer, ModelArgs
 
 
 
12
  from encoding_dsv4 import encode_messages, parse_message_from_completion_text
13
 
14
 
 
1
  import os
2
  import json
3
+ import sys
4
  from argparse import ArgumentParser
5
  from typing import List
6
 
 
10
  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
 
18