From 6071c0772bd21fb9161da9d5a18cf915ba1d2706 Mon Sep 17 00:00:00 2001 From: Sofia Liguori Date: Fri, 13 Jan 2023 11:46:08 +0000 Subject: [PATCH] Change addons requirements to https, disable wandb by default --- nlp/gpt_j/popxl/requirements.txt | 2 +- nlp/gpt_j/popxl/utils/setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/nlp/gpt_j/popxl/requirements.txt b/nlp/gpt_j/popxl/requirements.txt index cafd2b135..fd9fb0cfd 100644 --- a/nlp/gpt_j/popxl/requirements.txt +++ b/nlp/gpt_j/popxl/requirements.txt @@ -15,6 +15,6 @@ sklearn==0.0 pytest==6.2.5 pytest-pythonpath==0.7.4 -git+ssh://git@github.com/graphcore/popxl-addons.git@sdk-release-3.1_a +git+https://github.com/graphcore/popxl-addons.git@sdk-release-3.1_a protobuf==3.20.*; python_version > '3.6' diff --git a/nlp/gpt_j/popxl/utils/setup.py b/nlp/gpt_j/popxl/utils/setup.py index 015d699cf..f70cdd56a 100644 --- a/nlp/gpt_j/popxl/utils/setup.py +++ b/nlp/gpt_j/popxl/utils/setup.py @@ -124,7 +124,7 @@ def gptj_fine_tuning_setup( ) -> Tuple[GPTJConfig, argparse.Namespace, Optional[GPTJForCausalLM]]: """GPT-J setup for fine tunning scripts""" config, args, pretrained = gptj_config_setup(config_file, presets_key, default_config, - wandb_setup=True, hf_model_setup=True) + wandb_setup=False, hf_model_setup=True) return config, args, pretrained