wehub-resource-sync
cddb07a176
docs / deploy (push) Has been cancelled
docs / changes (push) Has been cancelled
docs / check-and-build (push) Has been cancelled
build container image / cpu (push) Has been cancelled
build container image / cuda (push) Has been cancelled
build container image / rocm (push) Has been cancelled
frontend checks / frontend-checks (push) Has been cancelled
frontend tests / frontend-tests (push) Has been cancelled
lfs checks / lfs-check (push) Has been cancelled
python checks / python-checks (push) Has been cancelled
python tests / py3.12: macos-default (push) Has been cancelled
python tests / py3.11: windows-cpu (push) Has been cancelled
python tests / py3.12: windows-cpu (push) Has been cancelled
python tests / py3.11: linux-cpu (push) Has been cancelled
typegen checks / typegen-checks (push) Has been cancelled
uv lock checks / uv-lock-checks (push) Has been cancelled
openapi checks / openapi-checks (push) Has been cancelled
python tests / py3.11: macos-default (push) Has been cancelled
python tests / py3.12: linux-cpu (push) Has been cancelled
21 lines
752 B
Python
21 lines
752 B
Python
from typing import Any
|
|
|
|
|
|
def get_flux_in_channels_from_state_dict(state_dict: dict[str | int, Any]) -> int | None:
|
|
"""Gets the in channels from the state dict."""
|
|
|
|
# "Standard" FLUX models use "img_in.weight", but some community fine tunes use
|
|
# "model.diffusion_model.img_in.weight". Known models that use the latter key:
|
|
# - https://civitai.com/models/885098?modelVersionId=990775
|
|
# - https://civitai.com/models/1018060?modelVersionId=1596255
|
|
# - https://civitai.com/models/978314/ultrareal-fine-tune?modelVersionId=1413133
|
|
|
|
keys = {"img_in.weight", "model.diffusion_model.img_in.weight"}
|
|
|
|
for key in keys:
|
|
val = state_dict.get(key)
|
|
if val is not None:
|
|
return val.shape[1]
|
|
|
|
return None
|