diffusers
bcc20e45 - Add Cosmos3 action generation support (#13823)

Commit
36 days ago
Add Cosmos3 action generation support (#13823) * Add Cosmos3 action generation support * Add README action examples * Use do_classifier_free_guidance property * Remove unused method * Add action policy example to pipelines doc * Adding model selection for action example doc. * Remove redundant casts * Rename _pack_action_tokens to _prepare_action_segment * Move validation checks to check_inputs * Add action arguments in the __call__ docstring * Move action mode check to check_inputs * Rename action to action_tokens * Add warning for num_frames ovewrite attempt * Rename action_tokens to raw_actions * Remove scheduler config override * Refactor action to use CosmosActionCondition * Fix examples script to support flow_shift arg * Apply styling fixes * Remove CosmosActionCondition properties, move to pipeline * Replace validate wiht post init * Set height/width/num_frames to None, raise error if set ofr action * Fix action_dim default setting * Remove video argument before v2v is added * Fix None args * Add _EMBODIMENT_TO_RAW_ACTION_DIM mapping * Remove --raw-action-dim from README.md * Added prompt upsampler docs and examples * Apply style fixes * Bugfix for action mrope compression factor * Add action prompt json formatting, update docs * Add hand_pose raw_action_dim value * Add missing args to docstrings --------- Co-authored-by: Atharva Joshi <atjoshi@nvidia.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com> Co-authored-by: YiYi Xu <yixu310@gmail.com>
Author
Parents
Loading