iris-image-gen / iris /model.py

Commit History

Fix conv2d bf16 crash on T4: iris/model.py
654d061
verified

asdf98 commited on

Fix: register context_proj as proper nn.Module (was lazy, not saved in checkpoints)
c98929a
verified

asdf98 commited on

Upload iris/model.py
88e5d09
verified

asdf98 commited on