huggingface/diffusers

Make `FlaxLMSDiscreteScheduler` jittable

Open

#2180 aperta il 31 gen 2023

Vedi su GitHub
 (8 commenti) (2 reazioni) (0 assegnatari)Python (4562 fork)batch import
good first issue

Metriche repository

Star
 (22.190 star)
Metriche merge PR
 (Merge medio 13g 1h) (96 PR mergiate in 30 g)

Descrizione

Please, see discussion in #2155. I was under the wrong assumption that FlaxLMSDiscreteScheduler had been made jittable as part of #1661, but it never was.

Guida contributor

Direzione di ricerca
Indaga l'implementazione corrente di FlaxLMSDiscreteScheduler nel repository diffusers. Rivedi la discussione nell'issue #2155 e la precedente PR #1661 per capire cosa è stato tentato. Esamina altri scheduler Flax che sono già jittable per comprenderne il pattern. Implementa la compilazione jit per lo scheduler, assicurandoti che tutte le operazioni siano compatibili con la trasformazione jit di JAX.
Tech stack
python
Dominio
performance
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-2 giorni
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Nuova
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
Basic knowledge of JAXFamiliarity with diffusers library
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
60