update

environment.yaml

@@ -1,38 +0,0 @@
-name: control
-channels:
-  - pytorch
-  - defaults
-dependencies:
-  - python=3.8.5
-  - pip=20.3
-  - cudatoolkit=11.3
-  - pytorch=1.13.1
-  - torchvision=0.14.1
-  - numpy=1.23.1
-  - pip:
-      - gradio==3.35.2
-      - albumentations==1.3.0
-      - opencv-contrib-python==4.3.0.36
-      - imageio==2.9.0
-      - imageio-ffmpeg==0.4.2
-      - pytorch-lightning==1.5.0
-      - omegaconf==2.1.1
-      - test-tube>=0.7.5
-      - streamlit==1.12.1
-      - einops==0.3.0
-      - webdataset==0.2.5
-      - kornia==0.6
-      - open_clip_torch==2.0.2
-      - invisible-watermark>=0.1.5
-      - streamlit-drawable-canvas==0.8.0
-      - torchmetrics==0.6.0
-      - timm==0.6.12
-      - addict==2.4.0
-      - yapf==0.32.0
-      - prettytable==3.6.0
-      - safetensors==0.2.7
-      - basicsr==1.4.2
-      - diffusers==0.17.1
-      - accelerate==0.17.0
-      - transformers==4.30.2
-      - xformers

requirements.txt

@@ -2,9 +2,7 @@
 torch==1.13.1+cu117
 torchvision==0.14.1+cu117
 torchaudio==0.13.1
-numpy==1.23.1
-gradio==3.35.2
-gradio-client==0.2.7
+numpy
 albumentations==1.3.0
 opencv-contrib-python
 imageio==2.9.0
@@ -31,3 +29,5 @@ accelerate==0.18.0
 transformers==4.30.2
 xformers==0.0.16
 triton
+gradio==3.35.2
+gradio-client==0.2.7