diff --git a/src/petals/models/bloom/config.py b/src/petals/models/bloom/config.py index a376aab..d6a8146 100644 --- 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 index f5dc6f6..dd5f6b1 100644 --- 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)