Add future annotations and ruff flag #43
Merged
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
* Add quantization of Mistral * Add quantization flag and add 8bit quantization * Better printing * Update quantization * Add pipeline * Update Mistral LM generation * Update LM quantization * Add unittests & update pipeline & prompt generation * Correct tests * Update version of Mistral, update docstrings & README.md * Format code
* feature: add batched prompt generation * feature: add --batch_size_prompt argument * test: add simple argument test * feature: add batched prompt generation speed measuring * refactor: remove redundant print * fix: change default batch_size_prompt to 64 * style: black formatting * refactor: typo
* feature: add batched image generation * test: modify image generation tests * fix: modify examples * fix: prompt objects weights * docs: update args description * docs: update prompt generation docstrings * [Automated] Updated coverage badge --------- Co-authored-by: GitHub Actions <[email protected]>
* Add batch annotation * Update docs & add test & fix batched annotation * Change default batch annotation * Fix annotation tests * Fix tests * [Automated] Updated coverage badge * Update annotation example & docstrings * Fix formatting * Fix docstring * [Automated] Updated coverage badge * refactor: replace annotate() with annotate_batch() * feature: replace owlv2 resize --------- Co-authored-by: Jan Cuhel <[email protected]> Co-authored-by: GitHub Actions <[email protected]> Co-authored-by: Nikita Sokovnin <[email protected]>
* feature: add SDXL-Lightning image generator * fix: minor example fix * feature: add prompt weighting to sdxl-lightning * docs: update README * test: add sdxl-lightning tests * [Automated] Updated coverage badge --------- Co-authored-by: GitHub Actions <[email protected]>
Co-authored-by: Jan Cuhel <[email protected]>
* Added padding & truncation to the processing * Format code * fix: change labels dtype to int64 * format: black --------- Co-authored-by: Jan Cuhel <[email protected]> Co-authored-by: Nikita Sokovnin <[email protected]>
|
Thanks, @justincdavis, for your contribution! Feel free to submit a separate PR for list/tuple renaming. If you have any other ideas for improvements, you can open an issue where we can discuss them. Merging to dev. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Solves the issues presented in #42, allows running datadreamer on older Python versions such as 3.8
closes #42
I could extend this to utilize only "modern" types by also changing the List/Tuple based type hints to list/tuple based etc.