From c9352879d3ff4d49c4f52910f57097fa7750d9e5 Mon Sep 17 00:00:00 2001 From: Harsha-Nori Date: Mon, 19 Aug 2024 13:34:47 -0700 Subject: [PATCH] Update pypi_upload.yml --- .github/workflows/pypi_upload.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/pypi_upload.yml b/.github/workflows/pypi_upload.yml index 981cf2334..87e62b765 100644 --- a/.github/workflows/pypi_upload.yml +++ b/.github/workflows/pypi_upload.yml @@ -159,7 +159,7 @@ jobs: name: Install guidance from wheels - run: pip install transformers torch name: Other installs - - run: python -c "import guidance; lm = guidance.models.Transformers('gpt2'); lm += '1,2,3,4,5,' + guidance.gen('num', max_tokens=5, temperature=0); print(f'\n{str(lm)=}\n'); assert lm['num'].startswith('6')" + - run: python -c "import guidance; import transformers; lm = guidance.models.Transformers('gpt2'); lm += '1,2,3,4,5,' + guidance.gen('num', max_tokens=5, temperature=0); print(f'\n Transformers Version:{transformers.__version__}\n\n{str(lm)=}\n'); assert lm['num'].startswith('6')" name: Run smoke test publish_wheels: