Require torch version <2.0.0 for demoDiffusion

Signed-off-by: Rajeev Rao <[email protected]>
hnkfwhw · Mar 31, 2023 · 03f2700 · 03f2700
1 parent a672582
commit 03f2700
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/demo/Diffusion/requirements.txt b/demo/Diffusion/requirements.txt
@@ -11,4 +11,5 @@ onnxruntime==1.14.1
 onnx-graphsurgeon==0.3.26
 polygraphy==0.44.2
 scipy
+torch<2.0.0
 transformers==4.26.1