Update tok_demo.py

tok_demo.py

@@ -54,7 +54,7 @@ model = model.to(device)
 model = model.eval()
 model.enable_nest = True
 
-@spaces.GPU(duration=1200)
+@spaces.GPU(duration=300)
 def viz_diff_slots(model, img, nums, cfg=1.0, return_img=False):
     img.to(device)
     n_slots_inf = []