Commit History

VAE: Add timestep conditioning
65dad79

Sapir commited on

Merge pull request #34 from LightricksResearch/add_atten_to_decoder
f63ea56
unverified

Yoav HaCohen commited on

Decoder: add AttentionResBlocks block
07ddecf

origordon commited on

Merge pull request #33 from LightricksResearch/compress-all-half-channels
427926d
unverified

Guy Shiran commited on

causal_video_autoencoder: add option to half channels in depth to space upsample block
d5e984f

guysrn commited on

Merge pull request #29 from LightricksResearch/scale_vae_decoder_v1
c4b2a35
unverified

origordon commited on

VAE Decoder: Inject noise between conv layers.
1f21780

origordon commited on

Merge pull request #30 from LightricksResearch/fix-no-flash-attention
05cb3e4
unverified

Sapir Weissbuch commited on

model: fix flash attention enabling - do not check device type at this point (can be CPU)
5940103

erichardson commited on

Feature: Add full bfloat16 support.
6a9d9a1

daniel shalem commited on

Merge pull request #19 from LightricksResearch/bfloat16-inference
91602f9
unverified

Sapir Weissbuch commited on

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

daniel shalem commited on

VAE: Support different latent_var_log options when returning intermediate features for 3D perceptual loss
7d89bb0

erichardson commited on

VAE: Support retuning intermediate features for 3d perceptual loss
028b6a1

erichardson 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