Merge pull request #388 from yuxumin/patch-2

Update imagefunc.py
2024-11-06 09:05:03 +08:00
parent 458456e464 b0b68c8194
commit edceb07294
1 changed files with 2 additions and 2 deletions
@@ -1626,7 +1626,7 @@ def get_a_person_mask_generator_model_path() -> str:
        import wget
        model_url = f'https://storage.googleapis.com/mediapipe-models/image_segmenter/selfie_multiclass_256x256/float32/latest/{model_name}'
        log(f"Downloading '{model_name}' model")
-        os.makedirs(model_file_path, exist_ok=True)
+        os.makedirs(os.path.dirname(model_file_path), exist_ok=True)
        wget.download(model_url, model_file_path)
    return model_file_path

@@ -2503,4 +2503,4 @@ minicpm_llama3_v25_prompts = """
        # OUTPUT
        Your output should just be an plain list of descriptions. No numbers, no extraneous labels, no hyphens.
        Create only one prompt.
-        """
+        """