huggingface/diffusers

Make `FlaxLMSDiscreteScheduler` jittable

Open

#2 180 ouverte le 31 janv. 2023

Voir sur GitHub
 (8 commentaires) (2 réactions) (0 assignés)Python (4 562 forks)batch import
good first issue

Métriques du dépôt

Stars
 (22 190 stars)
Métriques de merge PR
 (Merge moyen 13j 1h) (96 PRs mergées en 30 j)

Description

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.

Guide contributeur

Direction de recherche
Étudiez l'implémentation actuelle de FlaxLMSDiscreteScheduler dans le dépôt diffusers. Examinez la discussion dans l'issue #2155 et la PR précédente #1661 pour comprendre ce qui a été tenté. Consultez d'autres planificateurs Flax déjà jittable pour comprendre le modèle. Implémentez la compilation jit pour le planificateur, en veillant à ce que toutes les opérations soient compatibles avec la transformation jit de JAX.
Stack technique
python
Domaine
performance
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-2 jours
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Récente
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
Basic knowledge of JAXFamiliarity with diffusers library
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
60