meg-huggingface
commited on
Commit
·
fe8891f
1
Parent(s):
1bed6b0
just realtoxicityprompts
Browse files- src/envs.py +2 -2
src/envs.py
CHANGED
|
@@ -10,9 +10,9 @@ OWNER = "meg" # Change to your org - don't forget to create a results and reques
|
|
| 10 |
|
| 11 |
# For harness evaluations
|
| 12 |
DEVICE = "cuda:0" #if you add compute, for harness evaluations
|
| 13 |
-
LIMIT =
|
| 14 |
NUM_FEWSHOT = 0 # Change with your few shot for the Harness evaluations
|
| 15 |
-
TASKS_HARNESS = ["realtoxicityprompts"
|
| 16 |
|
| 17 |
# For lighteval evaluations
|
| 18 |
ACCELERATOR = "cpu"
|
|
|
|
| 10 |
|
| 11 |
# For harness evaluations
|
| 12 |
DEVICE = "cuda:0" #if you add compute, for harness evaluations
|
| 13 |
+
LIMIT = 1 # !!!! For testing, should be None for actual evaluations!!!
|
| 14 |
NUM_FEWSHOT = 0 # Change with your few shot for the Harness evaluations
|
| 15 |
+
TASKS_HARNESS = ["realtoxicityprompts"]#, "toxigen", "logiqa"]
|
| 16 |
|
| 17 |
# For lighteval evaluations
|
| 18 |
ACCELERATOR = "cpu"
|