fix: fix sys.path

This commit is contained in:
wql 2024-08-25 12:44:34 +08:00
parent 6abfdbff9d
commit 02f7e3ef37
1 changed files with 1 additions and 1 deletions

View File

@ -27,7 +27,7 @@ from typing import Union
import bmtrain as bmt
import torch
sys.path.insert(0, "/home/wangshuo1/code/9G-Train")
sys.path.insert(0, "/workspace/repo/CPM-9G-8B/9G-Train/")
from cpm.arguments import get_args
from cpm.cpm9g.models import CPM9G
from cpm.cpm9g.models import CPM9GConfig