Skip to content

Commit

Permalink
Update pypi_upload.yml
Browse files Browse the repository at this point in the history
  • Loading branch information
Harsha-Nori authored Aug 19, 2024
1 parent c46af80 commit c935287
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion .github/workflows/pypi_upload.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit c935287

Please sign in to comment.