Update inference/generate.py
Browse files- 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, '../
|
| 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 |
|