Update handler.py

handler.py

@@ -388,7 +388,8 @@ class EndpointHandler:
                 # Set random seeds
                 random.seed(config.seed)
                 np.random.seed(config.seed)
-                generator = torch.manual_seed(config.seed)
+                generator = torch.Generator(device='cuda')
+                torch.manual_seed(config.seed)
                 
                 # Prepare generation parameters for the video model (we omit params that are destined to Varnish, or things like the seed which is set externally)
                 generation_kwargs = {