inference/trillium/MaxDiffusion/SDXL is broken

[m-strzelczyk]

1. The setup commands: There's a git clone missing in one line, where we're supposed to clone the maxdiffusion repostiory. There's just an URL.
2. When I follow the installation steps, I get the following error after last pip install:

ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
tensorflow 2.15.0 requires ml-dtypes~=0.2.0, but you have ml-dtypes 0.5.3 which is incompatible.
Successfully installed jax-0.6.2 jaxlib-0.6.2 libtpu-0.0.17 ml_dtypes-0.5.3

3. When I try to run the final command, it gives this error:

(tpu) strzelczyk@t1v-n-7bd476b6-w-0:~/maxdiffusion$ LIBTPU_INIT_ARGS="--xla_tpu_rwb_fusion=false --xla_tpu_dot_dot_fusion_duplicated=true --xla_tpu_scoped_vmem_limit_kib=65536" python -m src.maxdiffusion.generate_sdxl src/maxdiffusion/configs/base_xl.yml run_name="my_run"
Traceback (most recent call last):
  File "/home/strzelczyk/miniconda3/envs/tpu/lib/python3.10/runpy.py", line 187, in _run_module_as_main
    mod_name, mod_spec, code = _get_module_details(mod_name, _Error)
  File "/home/strzelczyk/miniconda3/envs/tpu/lib/python3.10/runpy.py", line 110, in _get_module_details
    __import__(pkg_name)
  File "/home/strzelczyk/maxdiffusion/src/maxdiffusion/__init__.py", line 20, in <module>
    from .utils import (
  File "/home/strzelczyk/maxdiffusion/src/maxdiffusion/utils/__init__.py", line 38, in <module>
    from .dynamic_modules_utils import get_class_from_dynamic_module
  File "/home/strzelczyk/maxdiffusion/src/maxdiffusion/utils/dynamic_modules_utils.py", line 28, in <module>
    from huggingface_hub import HfFolder, cached_download, hf_hub_download, model_info
ImportError: cannot import name 'cached_download' from 'huggingface_hub' (/home/strzelczyk/miniconda3/envs/tpu/lib/python3.10/site-packages/huggingface_hub/__init__.py)

I assume there's a problem with the versioning of libraries.