From 4883923a66d9b8cb5b8c1512c47eb5cfb6b618c7 Mon Sep 17 00:00:00 2001 From: Aleksandr Borzunov Date: Wed, 6 Sep 2023 15:23:48 +0000 Subject: [PATCH] Show license link for Falcon-180B --- src/petals/models/falcon/config.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/petals/models/falcon/config.py b/src/petals/models/falcon/config.py index a1ae5e9..9fadede 100644 --- a/src/petals/models/falcon/config.py +++ b/src/petals/models/falcon/config.py @@ -31,6 +31,9 @@ class DistributedFalconConfig(DefaultRevisionMixin, FalconConfig, ClientConfig, def from_pretrained( cls, model_name_or_path: Union[str, os.PathLike, None], *args, dht_prefix: Optional[str] = None, **kwargs ): + if "180B" in model_name_or_path.upper(): + logger.info("Make sure you follow the Falcon-180B license: https://bit.ly/falcon-180b-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)