Show license links when loading models

bigscience-workshop · Jun 24, 2023 · baf56b4 · baf56b4
1 parent 47a2b1e
commit baf56b4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/petals/models/bloom/config.py b/src/petals/models/bloom/config.py
@@ -24,6 +24,8 @@ class DistributedBloomConfig(BloomConfig, SequenceManagerConfig, PTuneConfig, LM
     def from_pretrained(
         cls, model_name_or_path: Union[str, os.PathLike, None], *args, dht_prefix: Optional[str] = None, **kwargs
     ):
+        logger.info("Make sure you follow the BLOOM's terms of use: https://bit.ly/bloom-license")
+
         loading_from_repo = model_name_or_path is not None and not os.path.isdir(model_name_or_path)
         if loading_from_repo and dht_prefix is None:
             # We need "-petals" for backward compatibility with Petals < 1.2.0

diff --git a/src/petals/models/llama/config.py b/src/petals/models/llama/config.py
@@ -23,6 +23,11 @@ class DistributedLlamaConfig(LlamaConfig, SequenceManagerConfig, PTuneConfig, LM
     def from_pretrained(
         cls, model_name_or_path: Union[str, os.PathLike, None], *args, dht_prefix: Optional[str] = None, **kwargs
     ):
+        logger.info(
+            "LLaMA is available solely for non-commercial research purposes. "
+            "Make sure you follow the terms of use: https://bit.ly/llama-license"
+        )
+
         loading_from_repo = model_name_or_path is not None and not os.path.isdir(model_name_or_path)
         if loading_from_repo and dht_prefix is None:
             dht_prefix = str(model_name_or_path)