Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update starter model size estimates.

Browse Source
This commit is contained in:
Ryan Dick 2024-08-26 23:39:00 +00:00 committed by Brandon
parent cff382715a
commit 50085b40bb
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
invokeai/backend/model_manager/starter_models.py
View File

@ -94,7 +94,7 @@ STARTER_MODELS: list[StarterModel] = [
name="FLUX Schnell (Quantized)", name="FLUX Schnell (Quantized)",
base=BaseModelType.Flux, base=BaseModelType.Flux,
source="InvokeAI/flux_schnell::transformer/bnb_nf4/flux1-schnell-bnb_nf4.safetensors", source="InvokeAI/flux_schnell::transformer/bnb_nf4/flux1-schnell-bnb_nf4.safetensors",
description="FLUX schnell transformer quantized to bitsandbytes NF4 format. Total size with dependencies: ~14GB", description="FLUX schnell transformer quantized to bitsandbytes NF4 format. Total size with dependencies: ~12GB",
type=ModelType.Main, type=ModelType.Main,
dependencies=[t5_8b_quantized_encoder, flux_vae, clip_l_encoder], dependencies=[t5_8b_quantized_encoder, flux_vae, clip_l_encoder],
), ),
@ -102,7 +102,7 @@ STARTER_MODELS: list[StarterModel] = [
name="FLUX Dev (Quantized)", name="FLUX Dev (Quantized)",
base=BaseModelType.Flux, base=BaseModelType.Flux,
source="InvokeAI/flux_dev::transformer/bnb_nf4/flux1-dev-bnb_nf4.safetensors", source="InvokeAI/flux_dev::transformer/bnb_nf4/flux1-dev-bnb_nf4.safetensors",
description="FLUX dev transformer quantized to bitsandbytes NF4 format. Total size with dependencies: ~16GB", description="FLUX dev transformer quantized to bitsandbytes NF4 format. Total size with dependencies: ~12GB",
type=ModelType.Main, type=ModelType.Main,
dependencies=[t5_8b_quantized_encoder, flux_vae, clip_l_encoder], dependencies=[t5_8b_quantized_encoder, flux_vae, clip_l_encoder],
), ),
@ -118,7 +118,7 @@ STARTER_MODELS: list[StarterModel] = [
name="FLUX Dev", name="FLUX Dev",
base=BaseModelType.Flux, base=BaseModelType.Flux,
source="InvokeAI/flux_dev::transformer/base/flux1-dev.safetensors", source="InvokeAI/flux_dev::transformer/base/flux1-dev.safetensors",
description="FLUX dev transformer in bfloat16. Total size with dependencies: ~34GB", description="FLUX dev transformer in bfloat16. Total size with dependencies: ~33GB",
type=ModelType.Main, type=ModelType.Main,
dependencies=[t5_base_encoder, flux_vae, clip_l_encoder], dependencies=[t5_base_encoder, flux_vae, clip_l_encoder],
), ),