Colab Notebooks not working

[ThorJonsson]

Hi, unfortunately the colab notebooks are not working due to the environment dependencies not being fixed for all packages leading to dependency conflicts.

In the colab notebook:

%pip install -r requirements.txt
#%pip install -q -U --pre triton
#%pip install -q diffusers[torch]==0.11.1 transformers==4.26.0 bitsandbytes==0.35.4
#decord accelerate omegaconf einops ftfy gradio imageio-ffmpeg xformers

whether I comment out the use of the requirements file or comment out the pip installations neither actually works in compiling all the packages being specified.

These dependency conflicts result in the following issue when using the test script:

20:49:24.079709: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
WARNING The following values were not passed to accelerate launch and

No other information is given but the result directory is empty after this happens. What is confusing to me is that neither TensorRT or Tensorflow is something that was found in the requirements file.

How can I get these notebooks working and reproduce these results?

[fre-mont]

I also had this problem with "The following values were not passed to accelerate launch and" in colab demo, but I solve it as below;

from accelerate.utils import write_basic_config
write_basic_config()

Set this up and try running !accelerate launch test_fatezero.py --config=$CONFIG_NAME