psychedelicious
aaa6211625
chore(backend): ruff C420
2025-03-28 18:28:32 -04:00
Billy
a6b94e8ca4
Revert some files
2025-03-26 13:18:50 +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
Billy
64aa965160
Set ordering
2025-03-25 19:21:14 +11:00
Billy
d715c27d07
Add more known failures
2025-03-25 17:59:28 +11:00
Billy
515084577c
Test all imports work
2025-03-25 17:45:22 +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
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