Change unet_runner timestep input to int64

models/turbine_models/custom_models/sdxl_inference/unet_runner.py

@@ -152,7 +152,7 @@ def forward(
     sample = torch.rand(
         args.batch_size, 4, args.height // 8, args.width // 8, dtype=dtype
     )
-    timestep = torch.zeros(1, dtype=dtype)
+    timestep = torch.zeros(1, dtype=torch.int64)
     prompt_embeds = torch.rand(2 * args.batch_size, args.max_length, 2048, dtype=dtype)
     text_embeds = torch.rand(2 * args.batch_size, 1280, dtype=dtype)
     time_ids = torch.zeros(2 * args.batch_size, 6, dtype=dtype)