huggingface/diffusers

Make `FlaxLMSDiscreteScheduler` jittable

Open

#2.180 geöffnet am 31. Jan. 2023

Auf GitHub ansehen
 (8 Kommentare) (2 Reaktionen) (0 zugewiesene Personen)Python (4.562 Forks)batch import
good first issue

Repository-Metriken

Stars
 (22.190 Stars)
PR-Merge-Metriken
 (Durchschn. Merge 13T 1h) (96 gemergte PRs in 30 T)

Beschreibung

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.

Contributor Guide

Research-Richtung
Untersuche die aktuelle Implementierung von FlaxLMSDiscreteScheduler im diffusers Repository. Überprüfe die Diskussion in Issue #2155 und den vorherigen PR #1661, um zu verstehen, was versucht wurde. Sieh dir andere Flax Scheduler an, die bereits jit fähig sind, um das Muster zu verstehen. Implementiere die JIT Kompilierung für den Scheduler und stelle sicher, dass alle Operationen mit der JIT Transformation von JAX kompatibel sind.
Tech Stack
python
Domain
performance
Issue Type
Funktion
SchwierigkeitGeschätzte Implementierungsschwierigkeit für neue Contributors, von 1 für sehr kleine Änderungen bis 5 für Expertenarbeit.
3
Geschätzte ZeitEin grober Zeitrahmen, um zu recherchieren, zu implementieren, zu testen und einen Pull Request vorzubereiten.
1-2 Tage
AktivitätsstatusWie verfügbar das Issue gerade wirkt: frisch, aktiv, stale, blockiert oder wartet auf Maintainer-Input.
Frisch
KlarheitWie klar das Issue die erwartete Änderung, Akzeptanzkriterien und den nächsten Schritt erklärt.
Klar
Voraussetzungen
Basic knowledge of JAXFamiliarity with diffusers library
EinsteigerfreundlichkeitEin Score von 1 bis 100, der schätzt, wie zugänglich dieses Issue für First-Time Contributors ist.
60