Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Dev -> Main: New LM, fixed CPU usage, updated docs #31

Merged
merged 3 commits into from
Feb 5, 2024
Merged

Dev -> Main: New LM, fixed CPU usage, updated docs #31

merged 3 commits into from
Feb 5, 2024

Conversation

HonzaCuhel
Copy link
Contributor

@HonzaCuhel HonzaCuhel commented Feb 5, 2024

This PR includes:

  • Fix for CPU device (regarding this issue)
  • Add new LM prompt generator - TinyLlama
  • Updating docs: changing images' url

HonzaCuhel and others added 3 commits February 1, 2024 11:11
* Change moving the model

* Set max_memory

* Update

* Update

* Remove comments

---------

Co-authored-by: Jan Čuhel <[email protected]>
* Add support for TinyLlama

* Update

* Remove caption reads sentences

* [Automated] Updated coverage badge

* Update LM tests

* Add TinyLlama to README.md

* Change helper functions to private

---------

Co-authored-by: GitHub Actions <[email protected]>
Co-authored-by: Nikita Sokovnin <[email protected]>
Copy link

github-actions bot commented Feb 5, 2024

☂️ Python Coverage

current status: ✅

Overall Coverage

Lines Covered Coverage Threshold Status
727 356 49% 0% 🟢

New Files

File Coverage Status
datadreamer/prompt_generation/tinyllama_lm_prompt_generator.py 83% 🟢
TOTAL 83% 🟢

Modified Files

File Coverage Status
datadreamer/pipelines/generate_dataset_from_scratch.py 42% 🟢
datadreamer/prompt_generation/init.py 100% 🟢
datadreamer/prompt_generation/lm_prompt_generator.py 60% 🟢
datadreamer/prompt_generation/synonym_generator.py 31% 🟢
TOTAL 58% 🟢

updated for commit: 8dfaeb6 by action🐍

Copy link

github-actions bot commented Feb 5, 2024

Test Results

  6 files    6 suites   44m 3s ⏱️
 60 tests  25 ✅  35 💤 0 ❌
360 runs  150 ✅ 210 💤 0 ❌

Results for commit 8dfaeb6.

@HonzaCuhel HonzaCuhel changed the title Dev Dev -> Main: New LM, fixed CPU usage, updated docs Feb 5, 2024
Copy link
Member

@sokovninn sokovninn left a comment

Choose a reason for hiding this comment

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

LGTM

@sokovninn sokovninn assigned sokovninn and unassigned HonzaCuhel Feb 5, 2024
@sokovninn sokovninn merged commit 8da345b into main Feb 5, 2024
9 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants