pengunduhan model yang nyaman

Unduh Model Nyaman

Keterangan

Paket ini menyediakan cara sederhana untuk mengunduh model terlatih dari internet untuk digunakan dalam proyek Comfy Anda.

Instalasi

Untuk menginstal paket, Anda dapat menggunakan pip:

Versi pengujian:

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

Tambahkan dua baris berikutnya ke file persyaratan.txt Anda untuk versi pengujian:

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

Penggunaan

Diperlukan untuk menyediakan file yaml dengan model yang akan diunduh dan jalur untuk menyimpannya. Paket akan mengunduh model secara paralel menggunakan jumlah pekerja yang ditentukan.

Jika model harus diunduh dari GCS, maka diperlukan file kunci.

 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 ()

model.yaml

File models.yaml harus berada di root proyek dan harus memiliki struktur berikut:

 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

  ...

Jika model sudah diunduh dan Anda ingin memaksa pengunduhan, Anda dapat menggunakan kunci force_download .

Model dapat diunduh dari GCS atau dari internet melalui HTTP.

Lisensi

 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.