Merge pull request #1 from prusnak/convert-script-flake8

convert.py: fix flake8 issues + sort imports with isort
ggerganov · Apr 14, 2023 · 209e5f8 · 209e5f8
2 parents b9c372d + 54b7b12
commit 209e5f8
Show file tree

Hide file tree

Showing 2 changed files with 128 additions and 41 deletions.
diff --git a/convert-pth-to-ggml.py b/convert-pth-to-ggml.py
@@ -1,6 +1,8 @@
 # Compatibility stub
 
-import argparse, convert
+import argparse
+
+import convert
 
 parser = argparse.ArgumentParser(description='Convert a LLaMA model checkpoint to a ggml compatible file')
 parser.add_argument('dir_model',  help='directory containing the model checkpoint')