Yanxin Lu

AssertionError while inferencing with multiple gpus

2

I run the following command in the terminal and encountered assertion error. torchrun --standalone --nproc_per_node 3 scripts/inference.py configs/opensora/inference/16x512x512.py --ckpt-path OpenSora-v1-HQ-16x512x512.pth --prompt-path ./assets/texts/test_1.txt Traceback (most recent call last): File "/root/autodl-tmp/Open-Sora/scripts/inference.py", line...

AttributeError: module 'jax.random' has no attribute 'KeyArray'

15

I encountered the error while training the UNet. Training the UNet... Traceback (most recent call last): File "/content/diffusers/examples/dreambooth/train_dreambooth.py", line 18, in from diffusers import AutoencoderKL, DDPMScheduler, StableDiffusionPipeline, UNet2DConditionModel File "/usr/local/lib/python3.10/dist-packages/diffusers/__init__.py",...

flash-attn successfully installed but Flash Attention 2 is not available

6

cuda: 11.7 torch: 2.0.1 python: 3.10.9 release: flash_attn-2.3.5+cu117torch2.0cxx11abiFALSE-cp310-cp310-linux_x86_64.whl ``` File "/home/.conda/envs/venv310/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 566, in from_pretrained return model_class.from_pretrained( File "/home/.conda/envs/venv310/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3233, in from_pretrained config = cls._check_and_enable_flash_attn_2(config, torch_dtype=torch_dtype, device_map=device_map) File "/home/.conda/envs/venv310/lib/python3.10/site-packages/transformers/modeling_utils.py",...