adjust the resolution based on available VRAM. add elapsed time.

environment.yml

@@ -1,8 +1,6 @@
 # environment.yml
 name: storyboard-video
 channels:
-  - pytorch
-  - nvidia
   - conda-forge
   - defaults
 dependencies:
@@ -12,13 +10,13 @@ dependencies:
   # FFmpeg in env is easiest on Windows if you rely on conda-forge
   - ffmpeg=7.1
 
-  # PyTorch with CUDA runtime bundled (NOT using local CUDA 13.1 toolkit)
-  # Using compatible versions for PyTorch 2.5.1 with CUDA 12.4
-  - pytorch=2.5.1
-  - pytorch-cuda=12.4
-  - torchvision=0.20.1
-  - torchaudio=2.5.1
-
   # pip deps (mirrors requirements.txt)
   - pip:
+      # RTX 50xx (sm_120) requires CUDA 12.8 wheels
+      - --index-url https://download.pytorch.org/whl/cu128
+      - torch
+      - torchvision
+      - torchaudio
       - -r requirements.txt
+      - diffusers==0.36.0
+      - ftfy==6.3.1