diffusers
1d619934 - [Neuron] Add AWS Neuron (Trainium/Inferentia) as an officially supported device (#13289)

Commit
4 days ago
[Neuron] Add AWS Neuron (Trainium/Inferentia) as an officially supported device (#13289) * draft:add neuron as a legit backend * feat: neuron-specific changes in the pipeline * tests: eager tests * fix: style * fix:apr_02 beta * cleanup: remove tp part, for another pr * fix: restore ring_anything to ContextParallelConfig after over-aggressive TP cleanup The previous cleanup commit removed TensorParallelConfig (correct) but also accidentally removed ring_anything from ContextParallelConfig (incorrect). ring_anything is a context-parallel feature referenced in context_parallel.py and must remain in the config. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> * removal: style fix * tests:sdxl + flux2 * tests: simplify * tests: simplify * fix style * fix style * fix style for doc-builder * review: address comments * review:apply suggestion for the fix of index_for_timtestep * review: stronger guard on image slice * Apply suggestions from code review Co-authored-by: YiYi Xu <yixu310@gmail.com> * fix: when set_begin_index not implemented for scheduler * review:add maybe_adjust_dtype_for_device and apply to all models with downcasting needs * fix: dependency * Update src/diffusers/pipelines/stable_diffusion_xl/pipeline_stable_diffusion_xl.py Co-authored-by: YiYi Xu <yixu310@gmail.com> * review: apply maybe_adjust_dtype_for_device in pixart pipe * review: update .ai/models.md --------- Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com> Co-authored-by: YiYi Xu <yixu310@gmail.com>
Author
Parents
Loading