huggingface/diffusers

Make `FlaxLMSDiscreteScheduler` jittable

Aperta

#2180 aperta il 31 gen 2023

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

Metriche repository

Star
 (22.190 stelle)
Metriche merge PR
 (Merge medio 14g 10h) (101 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
Analizza l'implementazione di FlaxLMSDiscreteScheduler per identificare le operazioni incompatibili con la compilazione JIT di JAX (come flussi di controllo dinamici o forme non statiche) e rifattorizzale per renderle jittabili.
Tech stack
python
Dominio
machine learningai
Tipo issue
Bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
Mezza giornata
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
PythonJAX
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
40