huggingface/diffusers

Make `FlaxLMSDiscreteScheduler` jittable

Ouverte

#2 180 ouverte le 31 janv. 2023

Voir sur GitHubGuide contributeur
 (8 commentaires) (2 réactions) (0 personne assignée)Python (4 562 forks)batch import
good first issue

Métriques du dépôt

Stars
 (22 190 étoiles)
Métriques de merge PR
 (Merge moyen 14j 10h) (101 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
Examinez l'implémentation de FlaxLMSDiscreteScheduler pour identifier les opérations incompatibles avec la compilation JIT de JAX (telles que les flux de contrôle dynamiques ou les formes non statiques) et refactorisez les pour les rendre jittables.
Stack technique
python
Domaine
machine learningai
Type d'issue
Bug
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.
Une demi journée
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
PythonJAX
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
40