Commit History

Feature: Add mixed precision support and direct bfloat16 support.
1940326

daniel shalem commited on

Remove the word "pixart" from code.
d504563

Sapir commited on

VAE: Support more configurations for Encoder and Decoder blocks
43d3c68

erichardson commited on

Lint: added ruff.
325137b

Sapir commited on

VAE encode bugfix: remove bad dist_util.
eec4cb2

Sapir commited on

CausalVideoAutoencoder: made neater load_ckpt.
d699d2b

Sapir commited on

Ckpt conversion: script + usage examples updated.
bebbcd0

Sapir commited on

Merge pull request #7 from LightricksResearch/feature/fix-transformer-init-bug
fc02e02
unverified

Dudu Moshe commited on

transformer3d: init mode xora never happens because lower case needed.
a3498bb

dudumoshe commited on

import fixes.
b30014f

Sapir commited on

Added tpu flash attention.
4f52f00

Sapir commited on

working.
86b1a7e

Sapir commited on

refactor
c811a04

Sapir commited on