diff --git a/environment.yml b/environment.yml index daed11e..6b9c406 100644 --- a/environment.yml +++ b/environment.yml @@ -19,6 +19,11 @@ dependencies: - scipy - scikit-learn + - ftfy + - tqdm + - regex + + - tabulate - h5py - wandb - pyyaml @@ -29,6 +34,7 @@ dependencies: - einops - comet_ml - diffusers + - pip: - pykeops - nestargs