Update __init__.py

Add lama and bigscience lama

Update init.py
Add lama and bigscience lama
00903bc3 · KhalidAlt · GitHub · 6559ba0f · 00903bc3
Unverified Commit 00903bc3 authored Apr 28, 2022 by KhalidAlt Committed by GitHub Apr 28, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

lm_eval/tasks/__init__.py lm_eval/tasks/__init__.py +5 -2

No files found.
--- a/lm_eval/tasks/__init__.py
+++ b/lm_eval/tasks/__init__.py
@@ -324,8 +324,11 @@ TASK_REGISTRY = {
    "gem_xsum_challenge_test_covid": gem_xsum.GEMXSUMChallgeTestCovid,

   #LAMA
-    "lama-trex":lama.Trex,
-    "bigscience-lama":bigsciencelama.BigScienceLAMA,
+    "lama-trex": lama.Trex,
+    "lama-squad": lama.Squad,
+    "lama-google_re": lama.google_re,
+    "lama-concptnet": lama.Conceptnet,
+    "bigscience-lama":lama.BigScienceLAMA,
    # WinoBias
    "wino_bias_type1_pro": wino_bias.WinoBiasType1Pro,
    "wino_bias_type1_anti": wino_bias.WinoBiasType1Anti,