téléchargement du modèle confortable

Téléchargement du modèle confortable

Description

Ce package fournit un moyen simple de télécharger des modèles pré-entraînés depuis Internet à utiliser dans vos projets Comfy.

Installation

Pour installer le package, vous pouvez utiliser pip :

Version de test :

pip install -i https://test.pypi.org/simple/ comfy-model-download

Ajoutez les deux lignes suivantes à votre fichier exigences.txt pour la version de test :

 --index-url https://test.pypi.org/simple/
comfy_model_download

Usage

Il fallait fournir un fichier yaml avec les modèles à télécharger et le chemin pour les enregistrer. Le package téléchargera les modèles en parallèle en utilisant le nombre de travailleurs spécifié.

Si le modèle doit être téléchargé depuis GCS, il est nécessaire de fournir un fichier clé.

 from comfy_model_download import ModelDownload

download = ModelDownload (
    config_file = "models_config.yaml" ,
    max_workers = 2 ,
    model_path = "path/to/save/models" ,
    key_file = "path/to/key/file" ,
)
download . start ()

modèles.yaml

Le fichier models.yaml doit se trouver à la racine du projet et doit avoir la structure suivante :

 models :
  checkpoints :
    - name : JuggernautXL.safetensors
      url : gs://example_gcs_bucket/JuggernautXL.safetensors
      # force_download: true
    - name : AnimateLCM_sd15_t2v_lora.safetensors
      url : gs://example_gcs_bucket/AnimateLCM_sd15_t2v_lora.safetensors
      
  ipadapter :
    - name : ip-adapter-plus_sd15.safetensors
      url : gs://example_gcs_bucket/ip-adapter-plus_sd15.safetensors

  animatediff_models :
    - name : AnimateLCM_sd15_t2v.ckpt
      url : gs://example_gcs_bucket/AnimateLCM_sd15_t2v.ckpt

  ...

Si le modèle est déjà téléchargé et que vous souhaitez forcer le téléchargement, vous pouvez utiliser la clé force_download .

Les modèles peuvent être téléchargés depuis GCS ou depuis Internet via HTTP.

Licence

 MIT License

Copyright (c) 2024 Freepik Company, Ltd.

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in all
copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
SOFTWARE.