diffusion_baseline / envs /langflow.yml
jyliuAI's picture
Add files using upload-large-folder tool
784f572 verified
Raw History Blame Contribute Delete
570 Bytes
# Upstream specifies Python 3.12 and a cu124 wheel. The equivalent Conda
# pytorch/pytorch-cuda pair keeps all torch packages in one solver strategy.
name: dlb-langflow
channels: [pytorch, nvidia, conda-forge]
dependencies:
- python=3.12
- pip=24.2
- pytorch=2.5.1
- mkl<2024.1
- pytorch-cuda=12.4
- cuda-toolkit=12.4
- pip:
- datasets==3.6.0
- einops==0.8.1
- fsspec==2025.3.0
- huggingface-hub==0.36.0
- pyarrow==20.0.0
- pydantic==2.12.4
- pyyaml==6.0.2
- safetensors==0.5.3
- transformers==4.57.3