From 45f5fca38396c97f812c2c5e0567a563901f9ce2 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Tue, 21 Feb 2023 22:50:59 -0600 Subject: [PATCH] use correct fn to make dirs --- api/onnx_web/convert/diffusion/textual_inversion.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/onnx_web/convert/diffusion/textual_inversion.py b/api/onnx_web/convert/diffusion/textual_inversion.py index acd9331d..e4cfac27 100644 --- a/api/onnx_web/convert/diffusion/textual_inversion.py +++ b/api/onnx_web/convert/diffusion/textual_inversion.py @@ -1,4 +1,4 @@ -from os import mkdirs, path +from os import makedirs, path from huggingface_hub.file_download import hf_hub_download from transformers import CLIPTokenizer, CLIPTextModel from torch.onnx import export @@ -18,7 +18,7 @@ def convert_diffusion_textual_inversion(context: ConversionContext, name: str, b if path.exists(dest_path): logger.info("ONNX model already exists, skipping.") - mkdirs(path.join(dest_path, "text_encoder")) + makedirs(path.join(dest_path, "text_encoder")) embeds_file = hf_hub_download(repo_id=inversion, filename="learned_embeds.bin") token_file = hf_hub_download(repo_id=inversion, filename="token_identifier.txt")