]> git.djapps.eu Git - pkg/ggml/sources/ggml/commitdiff
starcoder : fix starcoder/convert-hf-to-ggml.py to handle current-folder output file...
authorBrad Ito <redacted>
Tue, 22 Aug 2023 09:34:07 +0000 (02:34 -0700)
committerGitHub <redacted>
Tue, 22 Aug 2023 09:34:07 +0000 (12:34 +0300)
examples/starcoder/convert-hf-to-ggml.py

index ef93dfde1443e3f9b52b7094b2f0d701f4b7ff8b..30af75cb9f8e5358bbf25fc8579882fc0b78dee5 100644 (file)
@@ -46,7 +46,9 @@ args = parser.parse_args()
 use_f16 = not args.use_f32
 
 fname_out = args.outfile
-os.makedirs(os.path.dirname(fname_out), exist_ok=True)
+fname_dir = os.path.dirname(fname_out)
+if fname_dir:
+    os.makedirs(fname_dir, exist_ok=True)
 
 print("Loading model: ", args.model_name_or_path)
 tokenizer = AutoTokenizer.from_pretrained(args.model_name_or_path)