mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
3534366146
Sometimes, diffusers model components (tokenizer, unet, etc.) have multiple weights files in the same directory. In this situation, we assume the files are different versions of the same weights. For example, we may have multiple formats (`.bin`, `.safetensors`) with different precisions. When downloading model files, we want to select only the best of these files for the requested format and precision/variant. The previous logic assumed that each model weights file would have the same base filename, but this assumption was not always true. The logic is revised score each file and choose the best scoring file, resulting in only a single file being downloaded for each submodel/subdirectory. |
||
|---|---|---|
| .. | ||
| libc_util.py | ||
| model_util.py | ||
| select_hf_files.py | ||