fix: use correct branch

feat: remove warning

fix(colab): don't preprocess images twice in CLIP

doc: simplify installation instructions

fix: use correct branch

fix: restore deleted paragraph

feat: link colab to correct branch

feat: remove warnings

feat: add tqdm

chore: reorganize files

Merge branch 'main' into chore-mv

feat(wandb-examples): use model file

Reorganization: move JAX VQGAN notebook to dev

Refactor: use VQGAN model from github, remove local copy

chore: reduce size of notebooks

feat: remove unused metrics

doc: note about model definition

feat: remove hardcoded values

refactor: move `captioned_strip` to library.

Get predictions from backend.

chore: move files around