yuntian-deng commited on
Commit
85bd4b9
·
1 Parent(s): 20a1b21

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -53,7 +53,7 @@ def draw_trace(image: np.ndarray, previous_actions: List[Tuple[str, List[int]]])
53
 
54
  # Initialize the model at the start of your application
55
  #model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
56
- model = initialize_model("config_identity.yaml", "yuntian-deng/computer-model")
57
 
58
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
59
  model = model.to(device)
 
53
 
54
  # Initialize the model at the start of your application
55
  #model = initialize_model("config_csllm.yaml", "yuntian-deng/computer-model")
56
+ model = initialize_model("2e5_debug_gpt_firstframe_posmap.yaml", "yuntian-deng/computer-model")
57
 
58
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
59
  model = model.to(device)