yuntian-deng commited on
Commit
8a16fb0
·
1 Parent(s): 5ecddc8

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -122,7 +122,7 @@ def draw_trace(image: np.ndarray, previous_actions: List[Tuple[str, List[int]]],
122
 
123
  # Initialize the model at the start of your application
124
  #model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
125
- model = initialize_model("pssearch_bsz64_acc1_lr8e5_512_leftclick.yaml", "yuntian-deng/computer-model")
126
 
127
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
128
  model = model.to(device)
 
122
 
123
  # Initialize the model at the start of your application
124
  #model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
125
+ model = initialize_model("pssearch_bsz64_acc1_lr8e5_512_leftclick_histpos_512_384.yaml", "yuntian-deng/computer-model")
126
 
127
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
128
  model = model.to(device)