.

environmental.yaml

@@ -1,121 +1,21 @@
 name: videograin
 channels:
   - pytorch
-  - nvidia
   - defaults
 dependencies:
-  - _libgcc_mutex=0.1=main
-  - _openmp_mutex=5.1=1_gnu
-  - blas=1.0=mkl
-  - brotli-python=1.0.9=py310h6a678d5_9
-  - bzip2=1.0.8=h5eee18b_6
-  - ca-certificates=2024.12.31=h06a4308_0
-  - certifi=2025.1.31=py310h06a4308_0
-  - charset-normalizer=3.3.2=pyhd3eb1b0_0
-  - cuda-cudart=12.1.105=0
-  - cuda-cupti=12.1.105=0
-  - cuda-libraries=12.1.0=0
-  - cuda-nvrtc=12.1.105=0
-  - cuda-nvtx=12.1.105=0
-  - cuda-opencl=12.8.55=0
-  - cuda-runtime=12.1.0=0
-  - cuda-version=12.8=3
-  - ffmpeg=4.3=hf484d3e_0
-  - filelock=3.13.1=py310h06a4308_0
-  - freetype=2.12.1=h4a9f257_0
-  - gmp=6.3.0=h6a678d5_0
-  - gmpy2=2.2.1=py310h5eee18b_0
-  - gnutls=3.6.15=he1e5248_0
-  - idna=3.7=py310h06a4308_0
-  - intel-openmp=2021.4.0=h06a4308_3561
-  - jinja2=3.1.5=py310h06a4308_0
-  - jpeg=9e=h5eee18b_3
-  - lame=3.100=h7b6447c_0
-  - lcms2=2.16=hb9589c4_0
-  - ld_impl_linux-64=2.40=h12ee557_0
-  - lerc=4.0.0=h6a678d5_0
-  - libcublas=12.1.0.26=0
-  - libcufft=11.0.2.4=0
-  - libcufile=1.13.0.11=0
-  - libcurand=10.3.9.55=0
-  - libcusolver=11.4.4.55=0
-  - libcusparse=12.0.2.55=0
-  - libdeflate=1.22=h5eee18b_0
-  - libffi=3.3=he6710b0_2
-  - libgcc-ng=11.2.0=h1234567_1
-  - libgomp=11.2.0=h1234567_1
-  - libiconv=1.16=h5eee18b_3
-  - libidn2=2.3.4=h5eee18b_0
-  - libjpeg-turbo=2.0.0=h9bf148f_0
-  - libnpp=12.0.2.50=0
-  - libnvjitlink=12.1.105=0
-  - libnvjpeg=12.1.1.14=0
-  - libpng=1.6.39=h5eee18b_0
-  - libstdcxx-ng=11.2.0=h1234567_1
-  - libtasn1=4.19.0=h5eee18b_0
-  - libtiff=4.5.1=hffd6297_1
-  - libunistring=0.9.10=h27cfd23_0
-  - libuuid=1.41.5=h5eee18b_0
-  - libwebp-base=1.3.2=h5eee18b_1
-  - llvm-openmp=14.0.6=h9e868ea_0
-  - lz4-c=1.9.4=h6a678d5_1
-  - mkl=2021.4.0=h06a4308_640
-  - mkl-service=2.4.0=py310h7f8727e_0
-  - mkl_fft=1.3.1=py310hd6ae3a3_0
-  - mkl_random=1.2.2=py310h00e6091_0
-  - mpc=1.3.1=h5eee18b_0
-  - mpfr=4.2.1=h5eee18b_0
-  - mpmath=1.3.0=py310h06a4308_0
-  - ncurses=6.4=h6a678d5_0
-  - nettle=3.7.3=hbbd107a_1
-  - networkx=3.4.2=py310h06a4308_0
-  - numpy=1.24.3=py310hd5efca6_0
-  - numpy-base=1.24.3=py310h8e6c178_0
-  - ocl-icd=2.3.2=h5eee18b_1
-  - openh264=2.1.1=h4ff587b_0
-  - openjpeg=2.5.2=he7f1fd0_0
-  - openssl=1.1.1w=h7f8727e_0
-  - pip=25.0=py310h06a4308_0
-  - pysocks=1.7.1=py310h06a4308_0
-  - python=3.10.0=h12debd9_5
-  - pytorch=2.3.1=py3.10_cuda12.1_cudnn8.9.2_0
-  - pytorch-cuda=12.1=ha16c6d3_6
-  - pytorch-mutex=1.0=cuda
-  - pyyaml=6.0.2=py310h5eee18b_0
-  - readline=8.2=h5eee18b_0
-  - requests=2.32.3=py310h06a4308_1
-  - setuptools=75.8.0=py310h06a4308_0
-  - six=1.16.0=pyhd3eb1b0_1
-  - sqlite=3.45.3=h5eee18b_0
-  - sympy=1.13.3=py310h06a4308_1
-  - tk=8.6.14=h39e8969_0
-  - torchaudio=2.3.1=py310_cu121
-  - torchtriton=2.3.1=py310
-  - torchvision=0.18.1=py310_cu121
-  - typing_extensions=4.12.2=py310h06a4308_0
-  - urllib3=2.3.0=py310h06a4308_0
-  - wheel=0.45.1=py310h06a4308_0
-  - xz=5.4.6=h5eee18b_1
-  - yaml=0.2.5=h7b6447c_0
-  - zlib=1.2.13=h5eee18b_1
-  - zstd=1.5.6=hc292b87_0
+  - ffmpeg=4.3
+  - numpy=1.24.3
+  - python=3.10.0
+  - pytorch=2.3.1
+  - pytorch-cuda=12.1
+  - pyyaml=6.0.2=
+  - torchvision=0.18.1
+  - yaml=0.2.5
   - pip:
       - absl-py==2.1.0
       - accelerate==0.24.1
-      - addict==2.4.0
-      - aiofiles==23.2.1
-      - aiohappyeyeballs==2.5.0
-      - aiohttp==3.11.13
-      - aiosignal==1.3.2
-      - altair==5.5.0
-      - annotated-types==0.7.0
       - antlr4-python3-runtime==4.9.3
-      - anyio==4.8.0
       - asttokens==2.4.1
-      - async-timeout==5.0.1
-      - attrs==25.1.0
-      - basicsr==1.4.2
-      - basicsr-fixed==1.4.2
       - bitsandbytes==0.35.4
       - click==8.1.7
       - coloredlogs==15.0.1
@@ -127,111 +27,58 @@ dependencies:
       - einops==0.8.0
       - exceptiongroup==1.2.1
       - executing==2.0.1
-      - fastapi==0.115.11
-      - ffmpy==0.5.0
       - flatbuffers==24.3.25
       - fonttools==4.53.1
-      - frozenlist==1.5.0
       - fsspec==2024.6.1
       - ftfy==6.2.0
-      - future==1.0.0
-      - gradio==3.44.4
-      - gradio-client==0.5.1
-      - groovy==0.1.2
       - grpcio==1.65.0
-      - h11==0.14.0
-      - httpcore==1.0.7
-      - httpx==0.28.1
-      - huggingface-hub==0.17.3
       - humanfriendly==10.0
       - imageio==2.34.2
       - imageio-ffmpeg==0.5.1
       - importlib-metadata==8.0.0
-      - importlib-resources==6.5.2
       - ipython==8.26.0
       - jedi==0.19.1
-      - joblib==1.4.2
-      - jsonschema==4.23.0
-      - jsonschema-specifications==2024.10.1
       - kiwisolver==1.4.5
-      - lazy-loader==0.4
-      - linkify-it-py==2.0.3
-      - lmdb==1.6.2
-      - loguru==0.7.3
       - markdown==3.6
-      - markdown-it-py==2.2.0
-      - markupsafe==2.1.5
       - matplotlib==3.9.1
       - matplotlib-inline==0.1.7
-      - mdit-py-plugins==0.3.3
-      - mdurl==0.1.2
       - modelcards==0.1.6
-      - multidict==6.1.0
-      - narwhals==1.29.1
-      - nltk==3.8.1
       - omegaconf==2.3.0
       - onnxruntime==1.18.1
       - onnxruntime-gpu==1.18.1
       - opencv-python==4.10.0.84
-      - orjson==3.10.15
       - packaging==24.1
-      - pandas==2.2.3
       - parso==0.8.4
       - pexpect==4.9.0
-      - pillow==10.4.0
-      - platformdirs==4.3.6
       - prompt-toolkit==3.0.47
-      - propcache==0.3.0
       - protobuf==4.25.3
       - psutil==6.0.0
       - ptyprocess==0.7.0
       - pure-eval==0.2.2
       - pyav==12.2.0
-      - pydantic==2.10.6
-      - pydantic-core==2.27.2
-      - pydub==0.25.1
       - pygments==2.18.0
       - pyparsing==3.1.2
       - python-dateutil==2.9.0.post0
-      - python-multipart==0.0.20
-      - pytz==2025.1
-      - rclone==0.4.4
-      - referencing==0.36.2
       - regex==2024.5.15
-      - rich==13.9.4
-      - rpds-py==0.23.1
-      - ruff==0.9.10
-      - safehttpx==0.1.6
       - safetensors==0.4.3
-      - scikit-image==0.24.0
-      - scikit-learn==1.2.2
-      - scipy==1.14.0
-      - semantic-version==2.10.0
-      - shellingham==1.5.4
-      - sniffio==1.3.1
+      - six==1.16.0
       - stack-data==0.6.3
-      - starlette==0.46.1
-      - tb-nightly==2.20.0a20250301
       - tensorboard==2.17.0
       - tensorboard-data-server==0.7.2
-      - threadpoolctl==3.5.0
-      - tifffile==2025.1.10
-      - timm==0.6.7
       - tokenizers==0.14.1
-      - tomli==2.2.1
-      - tomlkit==0.13.2
       - tqdm==4.66.4
       - traitlets==5.14.3
       - transformers==4.35.0
-      - typer==0.15.2
-      - tzdata==2025.1
-      - uc-micro-py==1.0.3
-      - uvicorn==0.34.0
+      - triton==2.3.1
       - wcwidth==0.2.13
-      - websockets==11.0.3
       - werkzeug==3.0.3
-      - xformers==0.0.27
-      - yapf==0.43.0
-      - yarl==1.18.3
       - zipp==3.19.2
-prefix: /home/xianyang/Data/env/anaconda3/envs/videograin
+      - scipy==1.14.0
+      - scikit-learn==1.2.2
+      - nltk==3.8.1
+      - timm==0.6.7
+      - scikit-image==0.24.0
+      - gdown==5.1.0
+      - basicsr-fixed
+      - huggingface-hub==0.17.3
+      - xformers==0.0.27

requirements.txt

@@ -66,4 +66,5 @@ timm==0.6.7
 scikit-image==0.24.0
 gdown==5.1.0
 basicsr-fixed
-huggingface-hub==0.17.3
+huggingface-hub==0.17.3
+xformers==0.0.27