Skip to content
This repository has been archived by the owner on Nov 3, 2023. It is now read-only.

Check directory depending on size param for gpt2 files #3158

Merged
merged 3 commits into from
Oct 7, 2020

Conversation

ankitade
Copy link
Contributor

@ankitade ankitade commented Oct 6, 2020

Patch description
Earlier distilgpt2 model was trying to load the gpt2 weights and hence failing during runtime. Now we load the files according to the type of the model being loaded

Testing steps
with-proxy python3 parlai/scripts/train_model.py -t integration_tests -m hugging_face/gpt2 -mf /tmp/model --max-train-time 100 -vmt ppl

os.path.isfile(
os.path.join(opt["datapath"], "models", "gpt2_hf", file_name)
)
PathManager.exists(os.path.join(hf_dir, file_name))
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lint seems to have caught a bug

@ankitade ankitade merged commit ebe7227 into master Oct 7, 2020
@ankitade ankitade deleted the distilgpt2-load branch October 7, 2020 19:23
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants