Spaces:
Runtime error
Runtime error
Commit
·
1948e30
1
Parent(s):
7741417
Update main.py
Browse files
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("
|
| 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)
|