update ver

app.py

@@ -11,6 +11,8 @@
 # // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # // See the License for the specific language governing permissions and
 # // limitations under the License.
+import spaces
+
 import os
 import torch
 import mediapy
@@ -34,8 +36,6 @@ from torchvision.transforms import Compose, Lambda, Normalize
 from torchvision.io.video import read_video
 import argparse
 
-import spaces
-
 from common.distributed import (
     get_device,
     init_torch,

requirements.txt

@@ -2,7 +2,7 @@
 einops==0.7.0                   # Tensor operations
 
 # Training
-torch==2.3.0                    # Torch
+torch==2.4.0                    # Torch
 omegaconf==2.3.0                # Configuration system.
 lpips==0.1.4                    # LPIPS model for VAE training
 
@@ -16,7 +16,7 @@ sentencepiece==0.2.0            # For Text Encoder
 diffusers==0.29.1               # Diffusers
 rotary-embedding-torch==0.5.3   # Rotary positional embedding
 transformers==4.38.2            # Transformers
-torchvision==0.18.0             # Torchvision
+torchvision==0.19.0             # Torchvision
 tiktoken==0.7.0                 # Tiktoken for generation
 transformers_stream_generator==0.0.5 #LLM generation support