fix PTQ4iT dependencies

cisco-open · Jan 15, 2025 · c93f448 · c93f448
1 parent 4cf2a84
commit c93f448
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/machine_learning_core/diffusion_model/PTQ4DiT/environment.yml b/machine_learning_core/diffusion_model/PTQ4DiT/environment.yml
@@ -2,11 +2,14 @@ name: modelsmith
 channels:
   - pytorch
   - nvidia
+  - conda-forge
+  - defaults
 dependencies:
-  - python >= 3.8
+  - python >=3.8
   - pytorch=2.1.0
   - torchvision=0.16.0
-  - pytorch-cuda=11.7
+  - torchaudio=2.1.0
+  - pytorch-cuda=11.8
   - pip
   - pip:
       - timm