Commit
·
24fab36
1
Parent(s):
09d65a8
add control experiments
Browse files
cognitive_mapping_probe/auto_experiment.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
import pandas as pd
|
| 2 |
import gc
|
|
|
|
| 3 |
from typing import Dict, List, Tuple
|
| 4 |
|
| 5 |
from .llm_iface import get_or_load_model
|
|
|
|
| 1 |
import pandas as pd
|
| 2 |
import gc
|
| 3 |
+
import torch
|
| 4 |
from typing import Dict, List, Tuple
|
| 5 |
|
| 6 |
from .llm_iface import get_or_load_model
|