15 Commits
Author SHA1 Message Date
Austin Mroz 8a9e5d56cb Remove errant on_prompt handler
The orchestration code included functionality to track when execution
had completed which is not useful outside of orchestration. As the other
references to completion had been removed, this code was causing
exceptions during execution.
2024-10-17 14:57:44 -05:00
Austin Mroz 535ea76a2a Extract just the execution code for publishing
A minimal change set was made to allow the code to function after
changes to the execution model. The changes unrelated to workflow
orchestration have been pulled out so that the code is viable for
general use once again.
2024-10-17 14:45:25 -05:00
Austin Mroz e03d06d959 Fix license reference 2024-08-01 13:35:39 -05:00
Austin Mroz f0aeafdd6f Merge pyproject and publish branches 2024-06-19 16:18:30 -05:00
snomiao 5c7c719195 chore(pyproject): Add pyproject.toml for Custom Node Registry 2024-06-19 07:00:13 +00:00
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