534 Commits

Author	SHA1	Message	Date
Billy	82dd2d508f	Deprecate checkpoint as file, diffusers as directory terminology	2025-03-27 08:10:12 +11:00
Billy	60b5aef16a	Log error -> warning	2025-03-27 06:56:22 +11:00
Billy	0e8b5484d5	Error handling	2025-03-26 19:31:57 +11:00
Billy	454506c83e	Type hints	2025-03-26 19:12:49 +11:00
Billy	8f6ab67376	Logs	2025-03-26 16:34:32 +11:00
Billy	5afcc7778f	Redundant	2025-03-26 16:32:19 +11:00
Billy	325e07d330	Error handling	2025-03-26 16:30:45 +11:00
Billy	a016bdc159	Add todo	2025-03-26 16:17:26 +11:00
Billy	a14f0b2864	Fail early on invalid config	2025-03-26 16:10:32 +11:00
Billy	721483318a	Extend ModelOnDisk	2025-03-26 16:10:00 +11:00
Billy	182580ff69	Imports	2025-03-26 12:55:10 +11:00
Billy	8e9d5c1187	Ruff formatting	2025-03-26 12:30:31 +11:00
Billy	99aac5870e	Remove star imports	2025-03-26 12:27:00 +11:00
jazzhaiku	ffa0beba7a	Merge branch 'main' into llava	2025-03-24 15:17:33 +11:00
psychedelicious	0b4c6f0ab4	fix(mm): flux model variant probing ... In #7780 we added FLUX Fill support, and needed the probe to be able to distinguish between "normal" FLUX models and FLUX Fill models. Logic was added to the probe to check a particular state dict key (input channels), which should be 384 for FLUX Fill and 64 for other FLUX models. The new logic was stricter and instead of falling back on the "normal" variant, it raised when an unexpected value for input channels was detected. This caused failures to probe for BNB-NF4 quantized FLUX Dev/Schnell, which apparently only have 1 input channel. After checking a variety of FLUX models, I loosened the strictness of the variant probing logic to only special-case the new FLUX Fill model, and otherwise fall back to returning the "normal" variant. This better matches the old behaviour and fixes the import errors. Closes #7822	2025-03-24 12:36:18 +11:00
Billy	d8450033ea	Fix	2025-03-21 17:46:18 +11:00
Billy	3938736bd8	Ruff formatting	2025-03-21 17:35:12 +11:00
Billy	fb2c7b9566	Defaults	2025-03-21 17:35:04 +11:00
Billy	29449ec27d	Implement new api for LLaVA	2025-03-21 17:17:56 +11:00
Billy	e38f778d28	Extend ModelOnDisk	2025-03-21 17:17:15 +11:00
Ryan Dick	5ea3ec5cc8	Get FLUX Fill working. Note: To use FLUX Fill, set guidance to ~30.	2025-03-19 14:45:18 +11:00
Ryan Dick	f7cfbd1323	Add FLUX Fill starter model.	2025-03-19 14:45:18 +11:00
Ryan Dick	2806b60701	Add logic to probe FLUX variant (NORMAL vs INPAINT).	2025-03-19 14:45:18 +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	b9972be7f1	Merge branch 'model-classification-api' into stripped-models	2025-03-18 14:57:23 +11:00
Billy	e61c5a3f26	Merge	2025-03-18 14:55:11 +11:00
Ryan Dick	9a389e6b93	Add a LLaVA OneVision starter model.	2025-03-18 11:53:06 +11:00
Ryan Dick	2ef1ecf381	Fix copy-paste errors.	2025-03-18 11:53:06 +11:00
Ryan Dick	e9714fe476	Add LLaVA Onevision model loading and inference support.	2025-03-18 11:53:06 +11:00
Ryan Dick	3f29293e39	Add LlavaOnevision model type and probing logic.	2025-03-18 11:53:06 +11:00
Billy	3469fc9843	Ruff	2025-03-18 09:22:16 +11:00
Billy	7cdd4187a9	Update classify script	2025-03-18 09:21:38 +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	8bcd9fe4b7	Extend ModelOnDisk	2025-03-17 09:18:51 +11:00
Billy	4377158503	Variant	2025-03-13 13:32:57 +11:00
Billy	d8b9a8d0dd	Merge branch 'main' into model-classification-api	2025-03-13 13:03:51 +11:00
Billy	39a4608d15	Fix annotations compatability 3.11	2025-03-13 13:01:19 +11:00
Billy	b86ac5e049	Explicit union	2025-03-13 10:28:07 +11:00
Billy	665236bb79	Type hints	2025-03-13 09:21:58 +11:00
Billy	f45400a275	Remove hash algo	2025-03-12 18:39:29 +11:00
psychedelicious	e35537e60a	fix(mm): move flux_redux starter model to the flux bundle, make siglip a dependency of it	2025-03-11 11:17:19 +11:00
Billy	d86b392bfd	Remove redundant hash_algo field	2025-03-11 09:16:59 +11:00
Billy	3e9e45b177	Update comments	2025-03-11 09:04:19 +11:00
Billy	907d960745	PR suggestions	2025-03-11 08:37:43 +11:00
Billy	bfdace6437	New API for model classification	2025-03-11 08:34:34 +11:00
psychedelicious	cf0cbaf0ae	chore: ruff (more)	2025-03-06 10:57:54 +11:00
psychedelicious	ac6fc6eccb	chore: ruff	2025-03-06 10:57:54 +11:00
Ryan Dick	8e28888bc4	Fix SigLipPipeline model size calculation.	2025-03-06 10:31:17 +11:00