This Pull Request fixes the error of the space

You need to import the `torchvision` library in your `requirements.txt`.

Click on _Merge_ to add this feature.

requirements.txt

@@ -13,3 +13,4 @@ torchsde==0.2.6
 einops
 opencv-contrib-python
 safetensors
+torchvision