Billy
182580ff69
Imports
2025-03-26 12:55:10 +11:00
Billy
8e14f9d971
Merge branch 'main' into stripped-models
2025-03-19 07:52:56 +11:00
psychedelicious
7fe4d4c21a
feat(app): better errors when scanning models with picklescan
...
Differentiate between malware detection and scan error.
2025-03-19 07:20:25 +11:00
Billy
24218b34bf
Make ruff happy
2025-03-17 12:04:26 +11:00
Billy
d970c6d6d5
Use override fixture
2025-03-17 11:58:13 +11:00
Billy
bfdace6437
New API for model classification
2025-03-11 08:34:34 +11:00
Jonathan
c094838c6a
Update model_util.py
2024-12-02 14:35:02 -08:00
Brandon Rising
ed46acee79
fix: Fail scan on InvalidMagicError in picklescan, update default for read_checkpoint_meta to scan unless explicitly told not to
2024-11-26 16:17:12 -05:00
Brandon Rising
c064257759
fix: Look in known subfolders for configs for clip variants
2024-11-07 12:01:02 +11:00
Brandon Rising
faa9ac4e15
fix: get_clip_variant_type should never return None
2024-11-06 09:59:50 +11:00
Brandon Rising
d07a6385ab
Always default to ClipVariantType.L instead of None
2024-11-05 12:03:40 -05:00
Brandon Rising
68df612fa1
fix: Never throw an exception when finding the clip variant type
2024-11-05 12:03:40 -05:00
Brandon Rising
b87f4e59a5
Create clip variant type, create new fucntions for discerning clipL and clipG in the frontend
2024-11-04 12:42:09 -05:00
Brandon Rising
0f333388bb
Add comment describing why we're not using the meta device during probing of gguf files
2024-10-02 18:33:05 -04:00
Ryan Dick
ec7e771942
Add a compute_dtype field to GGMLTensor.
2024-10-02 18:33:05 -04:00
Brandon Rising
7d9f125232
Run ruff and update imports
2024-10-02 18:33:05 -04:00
Brandon Rising
0875e861f5
Various updates to gguf performance
2024-10-02 18:33:05 -04:00
Brandon Rising
2bfb0ddff5
Initial GGUF support for flux models
2024-10-02 18:33:05 -04:00
Brandon Rising
a16b555d47
Simplify flux model dtype conversion in model loader
2024-09-05 15:47:14 -04:00
Brandon Rising
445f813fb9
Update flux transformer loader to more efficiently use and release memory during upcasting
2024-09-05 15:47:14 -04:00
Brandon Rising
33edee1ba6
Delete all flux bundle state dict keys when extracting the transformer state dict
2024-09-04 09:36:23 -04:00
Brandon Rising
d20335dabc
convert_bundle_to_flux_transformer_checkpoint now removes processed keys to decrease memory usage
2024-09-04 09:36:23 -04:00
Brandon Rising
d10d258213
Add a comment for why we're converting scale tensors in flux models to bfloat16
2024-09-04 09:36:23 -04:00
Brandon Rising
a005d06255
feat: support checkpoint bundles containing more than just the transformer
2024-09-04 09:36:23 -04:00
psychedelicious
5a3195f757
final tidying before marking PR as ready for review
...
- Replace AnyModelLoader with ModelLoaderRegistry
- Fix type check errors in multiple files
- Remove apparently unneeded `get_model_config_enum()` method from model manager
- Remove last vestiges of old model manager
- Updated tests and documentation
resolve conflict with seamless.py
2024-03-01 10:42:33 +11:00
Lincoln Stein
5d612ec095
Tidy names and locations of modules
...
- Rename old "model_management" directory to "model_management_OLD" in order to catch
dangling references to original model manager.
- Caught and fixed most dangling references (still checking)
- Rename lora, textual_inversion and model_patcher modules
- Introduce a RawModel base class to simplfy the Union returned by the
model loaders.
- Tidy up the model manager 2-related tests. Add useful fixtures, and
a finalizer to the queue and installer fixtures that will stop the
services and release threads.
2024-03-01 10:42:33 +11:00
