Commit History
Bugfix: pass device type for autocast.
d675fab
daniel shalem
commited on
vae: fix attention blocks and timestep conditioning
bd9f5fd
VAE: Check for timesteps parameter in decoder before calling
8920e2d
VAE: Add timestep conditioning
65dad79
Merge pull request #34 from LightricksResearch/add_atten_to_decoder
f63ea56
unverified
Decoder: add AttentionResBlocks block
07ddecf
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
Merge pull request #29 from LightricksResearch/scale_vae_decoder_v1
c4b2a35
unverified
VAE Decoder: Inject noise between conv layers.
1f21780
Merge pull request #26 from LightricksResearch/cuda_optional
7f1b6d2
unverified
Sapir Weissbuch
commited on
ran the ruff formatter
c042515
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
Added NIE on xla device
f5895e7
added cuda as optional
85a3cf8
README: added inference + installation guidelines, inference clearer.
b6c994f
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
VAE: Support retuning intermediate features for 3d perceptual loss
028b6a1
Remove the word "pixart" from code.
d504563
VAE: Support more configurations for Encoder and Decoder blocks
43d3c68
Lint: added ruff.
325137b
Image to video script: make determinist by random seed.
4bb89c5
Image to video: updated script.
4535a03
VAE encode bugfix: remove bad dist_util.
eec4cb2
Examples: update and fix scripts.
e46ff5e
CausalVideoAutoencoder: made neater load_ckpt.
d699d2b
Ckpt conversion: script + usage examples updated.
bebbcd0
Merge pull request #7 from LightricksResearch/feature/fix-transformer-init-bug
fc02e02
unverified
Dudu Moshe
commited on
