1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-10-27 21:33:02 +03:00
Files
diffusers/.ai
Jingya HUANG 1d61993458 [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>
2026-06-02 08:47:12 -10:00
..