yuntian-deng commited on
Commit
1948e30
·
1 Parent(s): 7741417

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -107,7 +107,7 @@ def draw_trace(image: np.ndarray, previous_actions: List[Tuple[str, List[int]]],
107
 
108
  # Initialize the model at the start of your application
109
  #model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
110
- model = initialize_model("2e5_debug_gpt_firstframe_posmap.yaml", "yuntian-deng/computer-model")
111
 
112
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
113
  model = model.to(device)
 
107
 
108
  # Initialize the model at the start of your application
109
  #model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
110
+ model = initialize_model("2e5_debug_gpt_firstframe_posmap_debugidentity.yaml", "yuntian-deng/computer-model")
111
 
112
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
113
  model = model.to(device)