channel inconsistency in cogvideo Lora training example #11114
Description
Describe the bug
while using the training script in (https://github.com/huggingface/diffusers/blob/main/examples/cogvideo/train_cogvideox_image_to_video_lora.py)
I made a dataset as described in readme and run training.
but a bug occurred at the forward pass process.It is because the model in-channel is 16 but model_input in-channel is 32.
how can i fix it?
Reproduction
# Sample noise that will be added to the latents
noise = torch.randn_like(video_latents)
# Add noise to the model input according to the noise magnitude at each timestep
# (this is the forward diffusion process)
noisy_video_latents = scheduler.add_noise(video_latents, noise, timesteps)
noisy_model_input = torch.cat([noisy_video_latents, image_latents], dim=2)
# Prepare rotary embeds
image_rotary_emb = (
prepare_rotary_positional_embeddings(
height=args.height,
width=args.width,
num_frames=num_frames,
vae_scale_factor_spatial=vae_scale_factor_spatial,
patch_size=model_config.patch_size,
attention_head_dim=model_config.attention_head_dim,
device=accelerator.device,
)
if model_config.use_rotary_positional_embeddings
else None
)
Predict the noise residual
model_output = transformer(
hidden_states=noisy_model_input,
encoder_hidden_states=prompt_embeds,
timestep=timesteps,
image_rotary_emb=image_rotary_emb,
return_dict=False,
)[0]
Logs
[rank0]: File "train_cogvideox_i_t2v_lora_raw.py", line 1426, in main
[rank0]: model_output = transformer(
[rank0]: ^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl
[rank0]: return self._call_impl(*args, **kwargs)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl
[rank0]: return forward_call(*args, **kwargs)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/parallel/distributed.py", line 1643, in forward
[rank0]: else self._run_ddp_forward(*inputs, **kwargs)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/parallel/distributed.py", line 1459, in _run_ddp_forward
[rank0]: return self.module(*inputs, **kwargs) # type: ignore[index]
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl
[rank0]: return self._call_impl(*args, **kwargs)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl
[rank0]: return forward_call(*args, **kwargs)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/accelerate/utils/operations.py", line 819, in forward
[rank0]: return model_forward(*args, **kwargs)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/accelerate/utils/operations.py", line 807, in __call__
[rank0]: return convert_to_fp32(self.model_forward(*args, **kwargs))
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/amp/autocast_mode.py", line 44, in decorate_autocast
[rank0]: return func(*args, **kwargs)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/diffusers/models/transformers/cogvideox_transformer_3d.py", line 476, in forward
[rank0]: hidden_states = self.patch_embed(encoder_hidden_states, hidden_states)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl
[rank0]: return self._call_impl(*args, **kwargs)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl
[rank0]: return forward_call(*args, **kwargs)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/diffusers/models/embeddings.py", line 715, in forward
[rank0]: image_embeds = self.proj(image_embeds)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1736, in _wrapped_call_impl
[rank0]: return self._call_impl(*args, **kwargs)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/modules/module.py", line 1747, in _call_impl
[rank0]: return forward_call(*args, **kwargs)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/modules/conv.py", line 554, in forward
[rank0]: return self._conv_forward(input, self.weight, self.bias)
[rank0]: ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank0]: File "/share/home/u21012/.conda/envs/snvds/lib/python3.11/site-packages/torch/nn/modules/conv.py", line 549, in _conv_forward
[rank0]: return F.conv2d(
[rank0]: ^^^^^^^^^
[rank0]: RuntimeError: Given groups=1, weight of size [1920, 16, 2, 2], expected input[6, 32, 60, 90] to have 16 channels, but got 32 channels insteadSystem Info
Name Version Build Channel
_libgcc_mutex 0.1 main
_openmp_mutex 5.1 1_gnu
absl-py 2.1.0 pypi_0 pypi
accelerate 1.4.0 pypi_0 pypi
addict 2.4.0 pypi_0 pypi
aiohappyeyeballs 2.5.0 pypi_0 pypi
aiohttp 3.11.13 pypi_0 pypi
aiosignal 1.3.2 pypi_0 pypi
aliyun-python-sdk-core 2.16.0 pypi_0 pypi
aliyun-python-sdk-kms 2.16.5 pypi_0 pypi
annotated-types 0.7.0 pypi_0 pypi
anyio 4.6.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
argon2-cffi 21.3.0 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
argon2-cffi-bindings 21.2.0 py311h5eee18b_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
asttokens 2.0.5 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
async-lru 2.0.4 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
attrs 24.3.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
babel 2.16.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
beautifulsoup4 4.12.3 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
blas 1.0 mkl https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
bleach 6.2.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
blinker 1.9.0 pypi_0 pypi
bottleneck 1.4.2 py311hf4808d0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
brotli-python 1.0.9 py311h6a678d5_9 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
bzip2 1.0.8 h5eee18b_6
c-ares 1.19.1 h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
ca-certificates 2025.2.25 h06a4308_0
cachetools 5.5.1 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
cairo 1.16.0 hb05425b_5 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
certifi 2025.1.31 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
cffi 1.17.1 py311h1fdaa30_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
charset-normalizer 3.3.2 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
click 8.1.8 pypi_0 pypi
comm 0.2.1 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
configargparse 1.7 pypi_0 pypi
contourpy 1.3.1 py311hdb19cb5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
crcmod 1.7 pypi_0 pypi
cryptography 44.0.2 pypi_0 pypi
cuda-cudart 12.1.105 0 nvidia
cuda-cupti 12.1.105 0 nvidia
cuda-libraries 12.1.0 0 nvidia
cuda-nvrtc 12.1.105 0 nvidia
cuda-nvtx 12.1.105 0 nvidia
cuda-opencl 12.8.90 0 nvidia
cuda-runtime 12.1.0 0 nvidia
cuda-version 12.8 3 nvidia
cycler 0.11.0 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
cyrus-sasl 2.1.28 h52b45da_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
cython 3.0.11 py311h5eee18b_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
dash 2.18.2 pypi_0 pypi
dash-core-components 2.0.0 pypi_0 pypi
dash-html-components 2.0.0 pypi_0 pypi
dash-table 5.0.0 pypi_0 pypi
datasets 3.3.2 pypi_0 pypi
dbus 1.13.18 hb2f20db_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
debugpy 1.8.11 py311h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
decorator 5.1.1 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
decord 0.6.0 pypi_0 pypi
defusedxml 0.7.1 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
descartes 1.1.0 pyhd3eb1b0_4 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
diffusers 0.32.2 pypi_0 pypi
dill 0.3.8 pypi_0 pypi
docker-pycreds 0.4.0 pypi_0 pypi
eigen 3.2.8 2 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
executing 0.8.3 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
expat 2.6.4 h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
ffmpeg 4.2.2 h20bf706_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
filelock 3.14.0 pypi_0 pypi
fire 0.4.0 pyh44b312d_0 esri
flask 3.0.3 pypi_0 pypi
fontconfig 2.14.1 h55d465d_3 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
fonttools 4.55.3 py311h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
freetype 2.12.1 h4a9f257_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
frozenlist 1.5.0 pypi_0 pypi
fsspec 2024.12.0 pypi_0 pypi
geos 3.10.6 h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
giflib 5.2.2 h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
gitdb 4.0.12 pypi_0 pypi
gitpython 3.1.44 pypi_0 pypi
glib 2.78.4 h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
glib-tools 2.78.4 h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
gmp 6.3.0 h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
gmpy2 2.2.1 py311h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
gnutls 3.6.15 he1e5248_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
graphite2 1.3.14 h295c915_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
grpcio 1.70.0 pypi_0 pypi
gst-plugins-base 1.14.1 h6a678d5_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
gstreamer 1.14.1 h5eee18b_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
h11 0.14.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
harfbuzz 4.3.0 hf52aaf7_2 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
hdf5 1.12.1 h2b7332f_3 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
httpcore 1.0.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
httpx 0.27.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
huggingface-hub 0.29.2 pypi_0 pypi
icu 73.1 h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
idna 3.7 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
importlib-metadata 8.6.1 pypi_0 pypi
intel-openmp 2023.1.0 hdb19cb5_46306 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
ipykernel 6.29.5 py311h06a4308_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
ipython 8.30.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
ipywidgets 8.1.5 pypi_0 pypi
itsdangerous 2.2.0 pypi_0 pypi
jedi 0.19.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jinja2 3.1.5 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jmespath 0.10.0 pypi_0 pypi
joblib 1.4.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jpeg 9e h5eee18b_3 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
json5 0.9.25 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jsonschema 4.23.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jsonschema-specifications 2023.7.1 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jupyter-lsp 2.2.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jupyter_client 8.6.3 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jupyter_core 5.7.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jupyter_events 0.12.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jupyter_server 2.15.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jupyter_server_terminals 0.4.4 py311h06a4308_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jupyterlab 4.3.4 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jupyterlab-widgets 3.0.13 pypi_0 pypi
jupyterlab_pygments 0.3.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
jupyterlab_server 2.27.3 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
kiwisolver 1.4.8 py311h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
krb5 1.20.1 h143b758_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
lame 3.100 h7b6447c_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
lcms2 2.16 hb9589c4_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
ld_impl_linux-64 2.40 h12ee557_0
lerc 4.0.0 h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libabseil 20240116.2 cxx17_h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libclang 14.0.6 default_hc6dbbc7_2 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libclang13 14.0.6 default_he11475f_2 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libcublas 12.1.0.26 0 nvidia
libcufft 11.0.2.4 0 nvidia
libcufile 1.13.1.3 0 nvidia
libcups 2.4.2 h2d74bed_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libcurand 10.3.9.90 0 nvidia
libcurl 8.12.1 hc9e6f67_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libcusolver 11.4.4.55 0 nvidia
libcusparse 12.0.2.55 0 nvidia
libdeflate 1.22 h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libedit 3.1.20230828 h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libev 4.33 h7f8727e_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libffi 3.4.4 h6a678d5_1
libgcc-ng 11.2.0 h1234567_1
libgfortran-ng 11.2.0 h00389a5_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libgfortran5 11.2.0 h1234567_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libglib 2.78.4 hdc74915_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libgomp 11.2.0 h1234567_1
libiconv 1.16 h5eee18b_3 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libidn2 2.3.4 h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libjpeg-turbo 2.0.0 h9bf148f_0 pytorch
libllvm14 14.0.6 hecde1de_4 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libnghttp2 1.57.0 h2d74bed_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libnpp 12.0.2.50 0 nvidia
libnvjitlink 12.1.105 0 nvidia
libnvjpeg 12.1.1.14 0 nvidia
libopus 1.3.1 h5eee18b_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libpng 1.6.39 h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libpq 17.4 hdbd6064_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libprotobuf 4.25.3 he621ea3_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libsodium 1.0.18 h7b6447c_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libssh2 1.11.1 h251f7ec_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libstdcxx-ng 11.2.0 h1234567_1
libtasn1 4.19.0 h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libtiff 4.5.1 hffd6297_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libunistring 0.9.10 h27cfd23_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libuuid 1.41.5 h5eee18b_0
libvpx 1.7.0 h439df22_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libwebp 1.3.2 h11a3e52_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libwebp-base 1.3.2 h5eee18b_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libxcb 1.15 h7f8727e_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libxkbcommon 1.0.1 h097e994_2 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
libxml2 2.13.5 hfdd30dd_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
llvm-openmp 14.0.6 h9e868ea_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
lz4-c 1.9.4 h6a678d5_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
markdown 3.7 pypi_0 pypi
markdown-it-py 3.0.0 pypi_0 pypi
markupsafe 3.0.2 py311h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
matplotlib 3.10.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
matplotlib-base 3.10.0 py311hbfdbfaf_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
matplotlib-inline 0.1.6 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
mdurl 0.1.2 pypi_0 pypi
mistune 3.1.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
mkl 2023.1.0 h213fc3f_46344 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
mkl-service 2.4.0 py311h5eee18b_2 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
mkl_fft 1.3.11 py311h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
mkl_random 1.2.8 py311ha02d727_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
motmetrics 1.1.3 py_0 esri
mpc 1.3.1 h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
mpfr 4.2.1 h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
mpmath 1.3.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
multidict 6.1.0 pypi_0 pypi
multiprocess 0.70.16 pypi_0 pypi
mysql 8.4.0 h29a9f33_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
narwhals 1.30.0 pypi_0 pypi
nbclient 0.10.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
nbconvert-core 7.16.6 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
nbformat 5.10.4 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
ncurses 6.4 h6a678d5_0
nest-asyncio 1.6.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
nettle 3.7.3 hbbd107a_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
networkx 3.4.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
notebook 7.3.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
notebook-shim 0.2.3 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
numexpr 2.10.1 py311h3c60e43_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
numpy 1.26.4 py311h08b1b3b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
numpy-base 1.26.4 py311hf175353_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
nuscenes-devkit 1.1.3 py_1 esri
ocl-icd 2.3.2 h5eee18b_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
open3d 0.18.0 pypi_0 pypi
opencv 4.10.0 py311hab19f7d_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
openh264 2.1.1 h4ff587b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
openjpeg 2.5.2 he7f1fd0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
openldap 2.6.4 h42fbc30_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
openssl 3.0.16 h5eee18b_0
openxlab 0.1.2 pypi_0 pypi
oss2 2.17.0 pypi_0 pypi
overrides 7.4.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
packaging 24.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pandas 2.2.3 py311h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pandocfilters 1.5.0 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
parso 0.8.4 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pcre2 10.42 hebb0a14_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
peft 0.14.0 pypi_0 pypi
pexpect 4.8.0 pyhd3eb1b0_3 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pillow 11.1.0 py311hcea889d_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pip 25.0 py311h06a4308_0
pixman 0.40.0 h7f8727e_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
platformdirs 3.10.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
plotly 6.0.0 pypi_0 pypi
ply 3.11 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
prometheus_client 0.21.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
prompt-toolkit 3.0.43 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
prompt_toolkit 3.0.43 hd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
propcache 0.3.0 pypi_0 pypi
protobuf 5.29.3 pypi_0 pypi
psutil 7.0.0 pypi_0 pypi
ptyprocess 0.7.0 pyhd3eb1b0_2 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pure_eval 0.2.2 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pyarrow 19.0.1 pypi_0 pypi
pycocotools 2.0.7 py311_2 esri
pycparser 2.21 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pycryptodome 3.22.0 pypi_0 pypi
pydantic 2.10.6 pypi_0 pypi
pydantic-core 2.27.2 pypi_0 pypi
pygments 2.15.1 py311h06a4308_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pyparsing 3.2.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pyqt 5.15.10 py311h6a678d5_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pyqt5-sip 12.13.0 py311h5eee18b_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pyquaternion 0.9.9 py_0 esri
pysocks 1.7.1 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
python 3.11.11 he870216_0
python-dateutil 2.9.0post0 py311h06a4308_2 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
python-fastjsonschema 2.20.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
python-json-logger 3.2.1 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
python-tzdata 2023.3 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pytorch 2.5.1 py3.11_cuda12.1_cudnn9.1.0_0 pytorch
pytorch-cuda 12.1 ha16c6d3_6 pytorch
pytorch-mutex 1.0 cuda pytorch
pytz 2023.4 pypi_0 pypi
pyyaml 6.0.2 py311h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
pyzmq 26.2.0 py311h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
qt-main 5.15.2 hb6262e9_12 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
readline 8.2 h5eee18b_0
referencing 0.30.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
regex 2024.11.6 pypi_0 pypi
requests 2.28.2 pypi_0 pypi
retrying 1.3.4 pypi_0 pypi
rfc3339-validator 0.1.4 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
rfc3986-validator 0.1.1 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
rich 13.4.2 pypi_0 pypi
rpds-py 0.22.3 py311h4aa5aa6_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
safetensors 0.5.3 pypi_0 pypi
scikit-learn 1.6.1 py311h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
scipy 1.15.1 py311h08b1b3b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
send2trash 1.8.2 py311h06a4308_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
sentencepiece 0.2.0 pypi_0 pypi
sentry-sdk 2.22.0 pypi_0 pypi
setproctitle 1.3.5 pypi_0 pypi
setuptools 60.2.0 pypi_0 pypi
shapely 1.8.5.post1 pypi_0 pypi
sip 6.7.12 py311h6a678d5_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
six 1.16.0 pyhd3eb1b0_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
smmap 5.0.2 pypi_0 pypi
sniffio 1.3.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
soupsieve 2.5 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
sqlite 3.45.3 h5eee18b_0
stack_data 0.2.0 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
sympy 1.13.1 pypi_0 pypi
tbb 2021.8.0 hdb19cb5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
tensorboard 2.19.0 pypi_0 pypi
tensorboard-data-server 0.7.2 pypi_0 pypi
termcolor 2.1.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
terminado 0.17.1 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
threadpoolctl 3.5.0 py311h92b7b1e_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
tinycss2 1.4.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
tk 8.6.14 h39e8969_0
tokenizers 0.21.0 pypi_0 pypi
torchaudio 2.5.1 py311_cu121 pytorch
torchtriton 3.1.0 py311 pytorch
torchvision 0.20.1 py311_cu121 pytorch
tornado 6.4.2 py311h5eee18b_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
tqdm 4.65.2 pypi_0 pypi
traitlets 5.14.3 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
transformers 4.49.0 pypi_0 pypi
typing-extensions 4.12.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
typing_extensions 4.12.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
tzdata 2025a h04d1e81_0
unicodedata2 15.1.0 py311h5eee18b_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
urllib3 1.26.20 pypi_0 pypi
wandb 0.19.8 pypi_0 pypi
wcwidth 0.2.5 pyhd3eb1b0_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
webencodings 0.5.1 py311h06a4308_1 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
websocket-client 1.8.0 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
werkzeug 3.0.6 pypi_0 pypi
wheel 0.45.1 py311h06a4308_0
widgetsnbextension 4.0.13 pypi_0 pypi
x264 1!157.20191217 h7b6447c_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
xmltodict 0.14.2 py311h06a4308_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
xxhash 3.5.0 pypi_0 pypi
xz 5.6.4 h5eee18b_1
yaml 0.2.5 h7b6447c_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
yarl 1.18.3 pypi_0 pypi
zeromq 4.3.5 h6a678d5_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
zipp 3.21.0 pypi_0 pypi
zlib 1.2.13 h5eee18b_1
zstd 1.5.6 hc292b87_0 https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
Who can help?
No response