huggingface/diffusers

Make `FlaxLMSDiscreteScheduler` jittable

Open

#2.180 aberto em 31 de jan. de 2023

Ver no GitHub
 (8 comments) (2 reactions) (0 assignees)Python (4.562 forks)batch import
good first issue

Métricas do repositório

Stars
 (22.190 stars)
Métricas de merge de PR
 (Mesclagem média 13d 1h) (96 fundiu PRs em 30d)

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.

Guia do colaborador

Direção de pesquisa
Investigue a implementação atual do FlaxLMSDiscreteScheduler no repositório diffusers. Revise a discussão na issue #2155 e na PR anterior #1661 para entender o que foi tentado. Examine outros agendadores Flax que já são jittable para entender o padrão. Implemente a compilação jit para o agendador, garantindo que todas as operações sejam compatíveis com a transformação jit do JAX.
Pilha de tecnologia
python
Domain
performance
Tipo Issue
Funcionalidade
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-2 dias
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Recente
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
Basic knowledge of JAXFamiliarity with diffusers library
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
60