fix: restore temporal sampling batch size expected by xportrait

core/test_xportrait.py

@@ -511,7 +511,7 @@ def visualize_mm(args, name, batch_data, infer_model, nSample, local_image_dir,
 
             gene_img, _ = infer_model.sample_log(
                 cond=c,
-                batch_size=nSample,
+                batch_size=args.num_drivings,
                 ddim=True,
                 ddim_steps=args.ddim_steps,
                 eta=args.eta,