585 Commits

Author	SHA1	Message	Date
Billy	d8450033ea	Fix	2025-03-21 17:46:18 +11:00
Billy	3938736bd8	Ruff formatting	2025-03-21 17:35:12 +11:00
Billy	f5e78436a8	Update regression test	2025-03-21 17:14:02 +11:00
Billy	6a15b5d9be	Add stripped models for testing llava	2025-03-21 15:34:20 +11:00
Billy	c8888be4c3	Formatting	2025-03-18 13:10:07 +11:00
Billy	db1aa38e98	Warning	2025-03-18 09:55:13 +11:00
Billy	12717d4a4d	Stripped model data	2025-03-18 09:51:10 +11:00
Billy	1953f3cbcd	Skip hashing to make test quicker	2025-03-18 09:50:18 +11:00
Billy	ad66c101d2	Remove stripped model files	2025-03-18 09:10:37 +11:00
Billy	f9f99f873d	More models	2025-03-17 04:18:44 +00:00
Billy	b1d336ce8a	Ruff	2025-03-17 12:19:27 +11:00
Billy	40c7be8f5d	Warning about missing test cases	2025-03-17 12:19:15 +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	e5308be0bb	Use override fixture	2025-03-17 11:31:20 +11:00
Billy	637b93d2d8	Ruff	2025-03-14 10:18:25 +11:00
Billy	565b160060	More tests	2025-03-14 10:17:43 +11:00
Billy	b86ac5e049	Explicit union	2025-03-13 10:28:07 +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	b9c7bc8b0e	chore: ruff	2025-03-07 11:45:49 +11:00
psychedelicious	0f45ee04a2	tests: fix test_extract_valid_metadata_from_image to accomodate prev commit	2025-03-07 11:45:49 +11:00
psychedelicious	8e46b03f09	tests: add tests for extract_metadata_from_image	2025-03-07 11:45:49 +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
Mary Hipp	201d7f1fdb	fix test	2025-03-06 10:41:47 +11:00
Mary Hipp	17a5b1bd28	fix test	2025-03-06 10:41:47 +11:00
Ryan Dick	f1fde792ee	Get FLUX Redux working: model loading and inference.	2025-03-06 10:31:17 +11:00
Billy	f2689598c0	Formatting	2025-03-06 09:11:00 +11:00
psychedelicious	d2f8db9745	tidy: remove unused utils	2025-03-06 07:49:35 +11:00
psychedelicious	c1cf01a038	tests: use dangerously_run_function_in_subprocess to fix configure_torch_cuda_allocator tests	2025-03-06 07:49:35 +11:00
psychedelicious	2bfb4fc79c	tests: add util to run a function in separate process ... This allows our tests to run in an isolated environment. For tests taht implicitly depend on import behaviour, this can prevent side-effects. The function should only be used for tests.	2025-03-06 07:49:35 +11:00
psychedelicious	d037d8f9aa	tests: update tests for configure_torch_cuda_allocator	2025-03-06 07:49:35 +11:00
psychedelicious	d5401e8443	tests: add testing utils to set/unset env var	2025-03-06 07:49:35 +11:00
Ryan Dick	49191709a0	Mark test_configure_torch_cuda_allocator_raises_if_torch_is_already_imported() to only run if CUDA is available.	2025-02-28 21:39:09 +00:00
Ryan Dick	a36a627f83	Switch from use_cuda_malloc flag to a general pytorch_cuda_alloc_conf config field that allows full customization of the CUDA allocator.	2025-02-28 21:39:09 +00:00
Ryan Dick	b31c71f302	Simplify is_torch_cuda_malloc_enabled() implementation and add unit tests.	2025-02-28 21:39:09 +00:00
psychedelicious	047c643295	tidy(app): document & clean up batch prep logic	2025-02-26 21:04:23 +11:00
psychedelicious	1bb8edf57e	perf(app): optimise batch prep logic even more ... Found another place where we deepcopy a dict, but it is safe to mutate. Restructured the prep logic a bit to support this. Updated tests to use the new structure.	2025-02-26 21:04:23 +11:00
psychedelicious	a3e78f0db6	perf(app): optimise batch prep logic ... - Avoid pydantic models when dict manipulation works - Avoid extraneous deep copies when we can safely mutate - Avoid NamedTuple construct and its overhead - Fix tests to use altered function signatures - Remove extraneous populate_graph function	2025-02-26 21:04:23 +11:00
psychedelicious	d66cd4e81b	tests(nodes): add test for collect -> iterate type validation	2025-02-06 07:57:26 +11:00
Ryan Dick	5357d6e08e	Rename ConcatenatedLoRALayer to MergedLayerPatch. And other minor cleanup.	2025-01-28 14:51:35 +00:00
Ryan Dick	28514ba59a	Update ConcatenatedLoRALayer to work with all sub-layer types.	2025-01-28 14:51:35 +00:00
Ryan Dick	206f261e45	Add utils for loading FLUX OneTrainer DoRA models.	2025-01-28 14:51:35 +00:00
Ryan Dick	dfa253e75b	Add utils for working with Kohya LoRA keys.	2025-01-28 14:51:35 +00:00
Ryan Dick	faa4fa02c0	Expand unit tests to test for confusion between FLUX LoRA formats.	2025-01-28 14:51:35 +00:00
Ryan Dick	5bd6428fdd	Add is_state_dict_likely_in_flux_onetrainer_format() util function.	2025-01-28 14:51:35 +00:00
Ryan Dick	8b4f411f7b	Add a test state dict for the OneTrainer DoRA format.	2025-01-28 14:51:35 +00:00