gauthiier/NATURESPEAK-ML-UTTER
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

revert demo

Browse Source
This commit is contained in:
gauthiier 2022-02-10 14:41:35 +01:00
parent b797b949af
commit 97bae0b66b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tokenise+train.py
View File

@ -15,7 +15,7 @@ def suffix(bs: int, ns: int, vs: int) -> str:
def train(filepath: str, ouputdir: Path, blocksize: int, vocabsize: int, num_steps: int, gpu: bool = False) -> str:
# from aitextgen.TokenDataset import TokenDataset
# from transformers import GPT2Config
from transformers import GPT2Config
from aitextgen.utils import build_gpt2_config
from aitextgen import aitextgen