Ryan Dick 3d4bd71098 Update test_probe_handles_state_dict_with_integer_keys() to make sure that it is still testing what it's intended to test. Previously, we were skipping an important part of the test by using a fake file path.

2024-10-02 18:33:05 -04:00

..

load

Add workaround for FLUX GGUF models with incorrect img_in.weight shape.

2024-10-02 18:33:05 -04:00

metadata

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

util

Add comment describing why we're not using the meta device during probing of gguf files

2024-10-02 18:33:05 -04:00

__init__.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

config.py

Initial GGUF support for flux models

2024-10-02 18:33:05 -04:00

merge.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

probe.py

Update test_probe_handles_state_dict_with_integer_keys() to make sure that it is still testing what it's intended to test. Previously, we were skipping an important part of the test by using a fake file path.

2024-10-02 18:33:05 -04:00

search.py

recognize .gguf files when scanning a folder for import

2024-10-02 18:33:05 -04:00

starter_models.py

Update starter model size estimates.

2024-08-26 20:17:50 -04:00