mirror
/
picoGPT
kopia lustrzana https://github.com/jaymody/picoGPT
1
0
Forkuj 0
Kod Zgłoszenia Packages Projekty Wydania Wiki Aktywność

Bug fix.

pull/4/head
jaymody 2023-01-28 19:43:17 +01:00
rodzic bf118a3660
commit 4cd64933bb
1 zmienionych plików z 1 dodań i 1 usunięć
Pokaż statystyki Ściągnij plik aktualizacji Ściągnij plik porównania Expand all files Collapse all files

2
main.py
Wyświetl plik

@ -83,10 +83,10 @@ def main(prompt, models_dir, model_size, n_tokens_to_generate):
assert model_size in ["124M", "355M", "774M", "1558M"] assert model_size in ["124M", "355M", "774M", "1558M"]
model_dir = os.path.join(models_dir, model_size) model_dir = os.path.join(models_dir, model_size)
tf_ckpt_path = tf.train.latest_checkpoint(model_dir)
if not os.path.isdir(model_dir): if not os.path.isdir(model_dir):
os.makedirs(model_dir) os.makedirs(model_dir)
download_gpt2_files(model_size, model_dir) download_gpt2_files(model_size, model_dir)
tf_ckpt_path = tf.train.latest_checkpoint(model_dir)
with open(os.path.join(model_dir, "hparams.json")) as file: with open(os.path.join(model_dir, "hparams.json")) as file:
hparams = json.load(file) hparams = json.load(file)