Commit Graph
10 Commits
Author SHA1 Message Date
snomiao db8f7a770c chore(publish): Add Github Action for Publishing to Comfy Registry 2024-06-19 07:00:12 +00:00
Austin Mroz 0839fbd33c Flesh out readme, Utilize client_id of comfy_runner 2024-06-18 15:14:55 -05:00
Austin Mroz 0375df5465 Successfully wrap the /prompt route
The implementation requires modifying frozen parameters, but I still
consider it by far preferable to needing the logic to select the
modified endpoint.
2024-06-18 14:29:32 -05:00
Austin Mroz 53cc59e535 Implement reset operation for net checkpoints
Fix duplicate prompt checks
2024-06-18 04:01:28 -05:00
Austin Mroz 8f34bf21dd Implement remote storage of checkpoints of S4
Working implementation of external checkpoints using Salad Simple Storage
Service
2024-06-17 22:06:10 -05:00
Austin Mroz 2d61e214b3 LICENSE, README, net fixes, SSS Specifics
Added LICENSE and brief readme

Updated network checking with more specifics to Salad Simple Storage.
The access token is now fetched, checkpoint size is checked before
uploading and checkpoints are read from the inputs folder to ease file
management on workflow dispatch
2024-06-15 15:31:16 -05:00
Austin Mroz 003ca04049 Implement checkpointing over http put requests
Aimed at targeting the Salad Simple Storage,
When net checkpointing is used, there is a system of priority to
requests and requests are uploaded asynchronously.
2024-06-15 00:17:18 -05:00
Austin Mroz 0ac9ad6fbe Save as single file, support cache completed nodes
Logic has been further cleaned up in preparation for networking
integration. There's not longer unique behaviour for what is being saved
and the onus is placed on the caller to provide data in a safetensor
appropriate format

Supported has been added to cache the output of completed nodes. This
code should be trivially expandable to non-sampler nodes, but will need
to be cautiously enabled on a per-node basis to better regulate dataflow
2024-06-14 18:02:41 -05:00
Austin Mroz 83dde78f0c Improved isolation of saving checkpoints.
In preperation for saving checkpoints remotely, all checkpoint
transactions are now wrapped.

Fixed step accumulation on consecutive failures

Support all builtin samplers
2024-06-14 04:25:50 -05:00
Austin Mroz 531fa3f9f2 Base checkpointing implementation
When sampling occurs, intermediate steps are cached to disk.
If at any point execution is halted, interrupted, or crashes, these
checkpoints are loaded from to restore execution from the previous step

Currently largely proof of concept. All samplers will cache, but
restoring is only applied to CustomSampler. Boilerplate is largely in
place to allow for caching output of other non-sampling nodes and will
be investigated later.
2024-06-14 02:40:50 -05:00