Commit graph

35 commits

Author SHA1 Message Date
Pierre Chapuis b2c26ab51c drop scipy dependency (now in bitsandbytes) 2024-01-31 15:03:06 +01:00
Pierre Chapuis 7eb8eb4c68 add support for pytorch 2.2 (2.1 is still supported)
also bump all dev dependencies to their latest version
2024-01-31 15:03:06 +01:00
Cédric Deltheil 146a1fae2a add mkdocstrings as dev dependency
Via:

    rye add --dev "mkdocstrings[python]"

Also added to requirements.docs.txt for GH Pages.
2024-01-31 14:27:47 +01:00
Pierre Chapuis 1eb5c779de better coverage config
- exclude e.g. tests
- include files not imported at all
- exclude some common patterns (see https://coverage.readthedocs.io/en/7.4.1/excluding.html#excluding-source-files)
2024-01-29 22:32:16 +01:00
Cédric Deltheil 5a8389ee0d pyproject.toml: formatting 2024-01-29 15:10:06 +01:00
Cédric Deltheil cd64bb29e6 add rye scripts for code coverage 2024-01-29 15:10:06 +01:00
Cédric Deltheil 1efb7a0e10 rye add --dev coverage 2024-01-29 15:10:06 +01:00
Bryce 12a5439fc4 refactor: rename noise => predicted_noise
and in euler, `alt_noise` can now be simply `noise`
2024-01-24 18:15:10 +01:00
Cédric Deltheil 9b9578cb7e add new rye script for mkdocs serving
i.e. useful for preview:

    rye run serve-docs
2024-01-08 17:54:54 +01:00
Cédric Deltheil f3f9ab8c3c add mkdocs-material as dev dependency
Via:

    rye add --dev mkdocs-material
2024-01-08 17:54:54 +01:00
limiteinductive 20c229903f upgrade pyright to 1.1.342 ; improve no_grad typing 2023-12-29 15:09:02 +01:00
limiteinductive 0f560437bc add tomli to training dependancy 2023-12-20 11:11:44 +01:00
Bryce 5ca1549c96 refactor: convert bash script to python
Ran successfully to completion. But on a repeat run `convert_unclip` didn't pass the hash check for some reason.

- fix inpainting model download urls
- shows a progress bar for downloads
- skips downloading existing files
- uses a temporary file to prevent partial downloads
- can do a dry run to check if url is valid `DRY_RUN=1 python scripts/prepare_test_weights.py`
- displays the downloaded file hash
2023-12-15 09:55:59 +01:00
Cédric Deltheil c8d5faff9b pyproject.toml: add example for combine-as-imports
b44d612 was misleading (s/avoid/allow/)
2023-12-11 13:57:57 +01:00
Cédric Deltheil b44d6122c4 pyproject.toml: enable isort rule in Ruff
Use `combine-as-imports = true` [1] to avoid such kinds or imports on a
single line:

    from torch import Tensor, device as Device, dtype as DType

[1]: https://docs.astral.sh/ruff/settings/#isort-combine-as-imports
2023-12-11 11:58:43 +01:00
Cédric Deltheil 4c07225d68 pyproject.toml: remove tool.isort section
Ruff is going to be used instead.

Follow up of #141
2023-12-11 11:58:43 +01:00
Cédric Deltheil 4fc5e427b8 training_utils: fix extra detection
Requirements could be, e.g.:

    wandb (>=0.15.7,<0.16.0) ; extra == "training"

Or:

    wandb>=0.16.0; extra == 'training'

Follow up of 86c5497
2023-12-08 19:09:16 +01:00
limiteinductive 86c54977b9 replace poetry by rye for python dependency management
Co-authored-by: Cédric Deltheil <cedric@deltheil.me>
Co-authored-by: Pierre Chapuis <git@catwell.info>
2023-12-08 17:40:10 +01:00
Pierre Chapuis f22f969d65 remove Black preview mode
also fix multiline logs in training
2023-12-04 14:15:56 +01:00
Cédric Deltheil e70dee987e use segment-anything unofficial Python package
Via:

    poetry remove segment-anything
    poetry add --optional segment-anything-py==1.0

Needed to publish on PyPI otherwise it fails (error: Can't have direct
dependency)
2023-10-21 13:51:06 +02:00
Cédric Deltheil e74454473d bump library version to v0.2.0 2023-10-20 18:28:31 +02:00
Cédric Deltheil e35dce825f pyproject.toml: add a note about scipy + bitsandbytes 2023-10-11 15:43:04 +02:00
Cédric Deltheil 5158187e96 poetry add torch@^2.1.0 2023-10-10 14:19:47 +02:00
Cédric Deltheil 9f3d064d14 poetry add --optional torchvision@^0.16.0 2023-10-10 14:19:47 +02:00
Cédric Deltheil 665bcdc95c add unit tests covering fluxion's gaussian_blur 2023-10-05 16:30:27 +02:00
Cédric Deltheil e352fea7c0 poetry add --optional diffusers@^0.21.0
T2I-Adapter for SDXL has been released part of v0.21.0
2023-09-25 13:54:26 +02:00
Pierre Chapuis b046f0cf3f update Pillow 2023-09-21 13:56:22 +02:00
Benjamin Trom 282578ddc0 add Segment Anything (SAM) to foundational models
Note: dense prompts (i.e. masks) support is still partial (see MaskEncoder)

Co-authored-by: Cédric Deltheil <cedric@deltheil.me>
2023-09-21 11:44:30 +02:00
Pierre Chapuis 0dfdd0d419 use extras instead of groups 2023-09-13 17:02:47 +02:00
Pierre Chapuis 0fd46f9ec4 make type checking strict 2023-08-17 14:44:45 +02:00
Benjamin Trom 663d7c414e Update pyproject.toml
Co-authored-by: Cédric Deltheil <355031+deltheil@users.noreply.github.com>
2023-08-17 14:44:45 +02:00
limiteinductive 89224c1e75 activate typing for all scripts 2023-08-17 14:44:45 +02:00
Cédric Deltheil 32425016c8 add missing torchvision dependency for training
Via:

    poetry add --optional torchvision
    poetry lock
2023-08-08 15:53:35 +02:00
Cédric Deltheil f49bb4f5fd pyproject.toml: bump pydantic to 2.0.3 2023-08-04 19:44:56 +02:00
Cédric Deltheil 48f674c433 initial commit 2023-08-04 15:28:41 +02:00