Correct model license metadata

jncraton · Jan 5, 2024 · e4e9bdc · e4e9bdc
1 parent 9c3f6ab
commit e4e9bdc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/changelog.md b/changelog.md
@@ -9,6 +9,7 @@
 - Updated to TinyLlama Chat v1.0
 - Remove auto model scaling on Colab
 - Correct phi-1.5 prompt format
+- Correct model license metadata
 
 ### Added
 

diff --git a/languagemodels/config.py b/languagemodels/config.py
@@ -74,7 +74,7 @@ class ModelFilterException(Exception):
         "quantization": "int8",
         "backend": "ct2",
         "architecture": "encoder-decoder-transformer",
-        "license": "cc-by-nc-4.0",  # HF says apache-2.0, but alpaca is NC
+        "license": "apache-2.0",
     },
     {
         "name": "flan-alpaca-gpt4-xl",
@@ -84,7 +84,7 @@ class ModelFilterException(Exception):
         "quantization": "int8",
         "backend": "ct2",
         "architecture": "encoder-decoder-transformer",
-        "license": "cc-by-nc-4.0",  # HF says apache-2.0, but alpaca is NC
+        "license": "apache-2.0",
     },
     {
         "name": "flan-t5-xl",
@@ -104,7 +104,7 @@ class ModelFilterException(Exception):
         "quantization": "int8",
         "backend": "ct2",
         "architecture": "encoder-decoder-transformer",
-        "license": "apache-2.0",  # This does use OpenAI-generated data
+        "license": "apache-2.0",
     },
     {
         "name": "LaMini-Flan-T5-783M",
@@ -144,7 +144,7 @@ class ModelFilterException(Exception):
         "quantization": "int8",
         "backend": "ct2",
         "architecture": "encoder-decoder-transformer",
-        "license": "cc-by-nc-4.0",  # HF says apache-2.0, but alpaca is NC
+        "license": "apache-2.0",
     },
     {
         "name": "flan-t5-base",