Update worker_runpod.py

worker_runpod.py

@@ -6,7 +6,7 @@ from diffusers.utils import export_to_video, load_image
 from transformers import T5EncoderModel, T5Tokenizer
 
 with torch.inference_mode():
-    model_id = "/content/model"
+    model_id = "/runpod-volume/model"
     
     # Load models and ensure they are placed on CUDA device
     transformer = CogVideoXTransformer3DModel.from_pretrained(model_id, subfolder="transformer", torch_dtype=torch.float16).to("cuda")