Skip to content

Commit

Permalink
flake8 fix
Browse files Browse the repository at this point in the history
  • Loading branch information
minimaxir committed May 30, 2020
1 parent 350626c commit aceafb2
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions aitextgen/aitextgen.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,12 +151,12 @@ def __init__(
)

os.rename(
os.path.join(cache_dir, f"pytorch_model.bin"),
os.path.join(cache_dir, "pytorch_model.bin"),
os.path.join(cache_dir, f"pytorch_model_{tf_gpt2}.bin"),
)

os.rename(
os.path.join(cache_dir, f"config.json"),
os.path.join(cache_dir, "config.json"),
os.path.join(cache_dir, f"config_{tf_gpt2}.json"),
)

Expand Down

0 comments on commit aceafb2

Please sign in to comment.