Open SumanthRH opened 7 months ago
Hi @jalammar , looks like I missed a commit on #110. All updates are on 68d893150d4fe81fcb34ebb39bf154d7aa437d6d . Changes since last time:
top_k_top_p_filtering
DeprecationWarning: invalid escape sequence '\.' layer_number = re.search("(?<=\.)\d+(?=\.)", name).group(0)
I've run pytest tests locally (python 3.10) and all tests pass. This should work now!
pytest tests
Ping @jalammar. I've also added a small test for bfloat16 and it works!
bfloat16
Also, I believe the pre-commit hooks need to be updated (had some issues), so I will add that in another PR.
Hi @jalammar , looks like I missed a commit on #110. All updates are on 68d893150d4fe81fcb34ebb39bf154d7aa437d6d . Changes since last time:
top_k_top_p_filtering
I've run
pytest tests
locally (python 3.10) and all tests pass. This should work now!