test run gen

app.py

@@ -306,7 +306,7 @@ def get_duration(rgb_image, seg_image, seed, randomize_seed,
                  cfg_interval_end, t_rescale):
     instance_labels = np.unique(np.array(seg_image).reshape(-1, 3), axis=0)
     step_duration = 15.0
-    return instance_labels.shape[0] * step_duration
+    return instance_labels.shape[0] * step_duration + 60
 
 @spaces.GPU(duration=get_duration)
 def run_generation(
@@ -873,7 +873,7 @@ if __name__ == '__main__':
 
     mogev2_id = 'Ruicheng/moge-2-vitl'
     moge_v2_dpt_model = MoGeModel.from_pretrained(mogev2_id).to(
-        "cpu", dtype=torch.float32
+        "cpu"
     )
 
     rot = np.array([