da03 commited on
Commit
292dd30
·
1 Parent(s): d0185ca
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -383,7 +383,7 @@ async def websocket_endpoint(websocket: WebSocket):
383
  next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
384
  # Load and append the corresponding ground truth image instead of model output
385
  print ('here4', len(previous_frames))
386
- if True and DEBUG_TEACHER_FORCING:
387
  img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
388
  previous_frames.append(img)
389
  elif False:
 
383
  next_frame, next_frame_append = predict_next_frame(previous_frames, previous_actions)
384
  # Load and append the corresponding ground truth image instead of model output
385
  print ('here4', len(previous_frames))
386
+ if False and DEBUG_TEACHER_FORCING:
387
  img = Image.open(f"record_100/image_{82+len(previous_frames)}.png")
388
  previous_frames.append(img)
389
  elif False: