da03 commited on
Commit
9725fa4
·
1 Parent(s): dfecf95
Files changed (1) hide show
  1. main.py +3 -0
main.py CHANGED
@@ -98,6 +98,9 @@ def prepare_model_inputs(
98
  # Clamp coordinates to valid ranges
99
  x = min(max(0, x), SCREEN_WIDTH - 1) if x is not None else 0
100
  y = min(max(0, y), SCREEN_HEIGHT - 1) if y is not None else 0
 
 
 
101
 
102
  inputs = {
103
  'image_features': previous_frame.to(device),
 
98
  # Clamp coordinates to valid ranges
99
  x = min(max(0, x), SCREEN_WIDTH - 1) if x is not None else 0
100
  y = min(max(0, y), SCREEN_HEIGHT - 1) if y is not None else 0
101
+ if DEBUG_MODE:
102
+ print ('DEBUG MODE, SETTING TIME STEP TO 0')
103
+ time_step = 0
104
 
105
  inputs = {
106
  'image_features': previous_frame.to(device),