Commit Graph

34 Commits

Author SHA1 Message Date
torzdf
9cda7aacae
Bugfix: requirements, pin opencv to prevent numpy2.x install 2025-07-11 18:19:27 +01:00
torzdf
5212589242 Pin imageio-ffmpeg to <0.6.0 2025-05-19 16:48:50 +01:00
torzdf
6fe300e601 pin numpy to < 2.0 2024-08-02 18:42:57 +01:00
torzdf
23944c3c10 Update requirements 2024-03-12 14:42:23 +00:00
torzdf
6a3b4ed4b3 Training fixes:
- Fix mixed precision layer storing and switching
  - typofix in _requirements_base.txt
  - lib.model.layers - cleanup, docs and tests fixes
  - lib.model.nn_blocks - typing fixes
2023-07-08 16:47:25 +01:00
torzdf
efbbe55267 bugfix: Pillow version pinning 2023-07-06 11:41:17 +01:00
torzdf
6a3b674bef
Rebase code (#1326)
* Remove tensorflow_probability requirement

* setup.py - fix progress bars

* requirements.txt: Remove pre python 3.9 packages

* update apple requirements.txt

* update INSTALL.md

* Remove python<3.9 code

* setup.py - fix Windows Installer

* typing: python3.9 compliant

* Update pytest and readthedocs python versions

* typing fixes

* Python Version updates
  - Reduce max version to 3.10
  - Default to 3.10 in installers
  - Remove incompatible 3.11 tests

* Update dependencies

* Downgrade imageio dep for Windows

* typing: merge optional unions and fixes

* Updates
  - min python version 3.10
  - typing to python 3.10 spec
  - remove pre-tf2.10 code
  - Add conda tests

* train: re-enable optimizer saving

* Update dockerfiles

* Update setup.py
  - Apple Conda deps to setup.py
  - Better Cuda + dependency handling

* bugfix: Patch logging to prevent Autograph errors

* Update dockerfiles

* Setup.py - Setup.py - stdout to utf-8

* Add more OSes to github Actions

* suppress mac-os end to end test
2023-06-27 11:27:47 +01:00
torzdf
03f5c671bc
Remove plaidML support (#1325)
* Remove PlaidML reference from readme files

* Remove AMD option from installers

* remove amd requirements and update setup.py

* remove plaidml test from CI workflow

* gpustats: remove plaidml backend

* plaid removals:
  - faceswap.py - python version check
  - setup.cfg - plaidml typing ignore
  - lib.keras_utils - All plaid code
  - lib.launcher.py - All plaidml checks and configuration

* remove tf2.2 specific code from GUI event reader

* lib.model - remove all plaidml implementations

* plugins.extract - remove plaidml code

* plugins.train remove plaidml code

* lib.convert - remove plaidml code

* tools.model: remove plaidml code

* Remove plaidML tests from unit tests

* remove plaidml_utils and docsting cleanups

* Remove plaidML refs from configs

* fix keras imports
2023-06-21 12:57:33 +01:00
torzdf
b4212dedc9 bugfix: pin numpy 2023-02-03 12:20:42 +00:00
torzdf
4f79ea47e6 bugfix: macos - raise tensorflow-metal max version
typofix: usage.md
2023-01-19 18:02:50 +00:00
torzdf
c455601ed2 Linux - ROCm (AMD) support 2023-01-11 13:09:32 +00:00
torzdf
dc94ed9cad
DirectML Support (#1291)
* DML Support
  - Add requirements. Update setup.py. Update Windows Installer
  - Add 'directml' backend. Update backend supporting logic
  - GPUStats - Add GPU querying from Win SDK for DML backend
  - Windows PlaidML - Deprecation Warning
  - Remove deprecated '-d' training flag
  - Logging in DML functions
- Unittest
  - Windows Unit Test
  - lib.utils
2023-01-06 14:45:57 +00:00
torzdf
d750af1e92 bump TF to version 2.10 2022-12-20 15:30:26 +00:00
torzdf
eefffe243d Bugfix: AMD. Pin numpy, matplotlib and numexpr 2022-12-20 01:41:03 +00:00
torzdf
01b03209dd
Update _requirements_base.txt
Update requirements.txt - pin matplotlib
2022-12-15 22:58:05 +00:00
torzdf
2beceffad9
Data Augmentation update (#1263)
- lib.detected_face
    - Subclass Masks for Landmark based masks
    - Add training mask propery + methods to DetectedFace
  - lib.training_training
    - subclass TrainingDataGenerator for training and preview data
    - Split cache into own module
    - Reduce thread count to 1 to prevent image corruption + data re-use
    - Process on largest model input/output size rather than stored image size
    - Size and crop masks during caching stage
    - Implement ring buffer for data flow
    - Fix preview reload bug
  - augmentation
    - typing
    - switch color aug order
    - better initialization
    - Fix warp + landmark warp to correctly apply at different image scales
    - Slightly improved warp caching
    - Don't store whether image is_preview. Handle all data as training images implicitly
  - plugins.trainer: Typing and fixes to work with trainingdata refactor
2022-08-21 18:59:45 +01:00
torzdf
39ac1f53a3 setup.py - Fix AMD conflicts 2022-08-10 17:38:04 +01:00
torzdf
6cd30126fd Revert "pin max numpy version to 1.22.x"
This reverts commit 997db773c3.
2022-08-03 00:57:47 +01:00
torzdf
997db773c3 pin max numpy version to 1.22.x 2022-08-03 00:37:19 +01:00
torzdf
c79ed7628a re-pin Tensorflow Probability 2022-08-02 18:33:45 +01:00
torzdf
3b95c3c9c2 Update python 3.7 pins 2022-08-02 17:33:43 +01:00
torzdf
92f142b8e7 Lower numpy + scikit-image pins for python 3.7 2022-08-02 17:25:58 +01:00
torzdf
26e26c6288 Update TF to 2.9
- Update TF to 2.7 to 2.9
    - Bump dependencies
    - Remove decode from pynvml calls
    - force keras predict functions to non-verbose
    - update tests
    - update Tensorboard logging
    - Update docs
2022-08-02 17:12:26 +01:00
torzdf
03f6cb4e7e setup.py: implement logging 2022-07-28 23:53:31 +01:00
torzdf
cfad829999 Update typing-extensions pin 2022-06-26 16:34:50 +01:00
torzdf
8ec1e1a091 Nvidia: Add AutoClip training option 2022-06-23 18:00:22 +01:00
torzdf
4f49de52e7 pin protobuf
Setup: automatically detect for apple-silicon
2022-06-16 16:40:49 +01:00
geewiz94
0f67790071 Some Apple Silicion setup fixes (#1238)
* Move pynvx to Nvidia requirements

* Add missing tensorflow-deps and fix tensorflow-metal version

* Fix setup not using Conda

* Update INSTALL.md
2022-06-16 16:40:49 +01:00
torzdf
a586ef6bf3 Add Apple M1 to setup.py
add libblas to requirements
2022-06-16 16:40:45 +01:00
torzdf
20a657d6cf Bump minimum TF Version 2022-06-07 10:54:51 +01:00
torzdf
c6ac90dea9 require typing-extensions 2022-06-05 23:01:15 +01:00
torzdf
9168721e42 AMD Updates
- Windows Installer - Default to Python3.8
   - Requirements - Pin protobuf to 3.19
2022-06-05 12:25:03 +01:00
torzdf
dbcd507d01 pin nvidia-ml-py for breaking change 2022-05-19 10:11:18 +01:00
torzdf
17d35ea423 setup.py - python 3.9 support
- move requirements files to dedicated folder
    - update all references to requirements files locations
    - setup.py: remove git requirements references
    - setup.py: allow running in python 3.9
    - windows/linux installers: default to python 3.9 environment
    - faceswap.py: cleaner python version check
2022-05-16 11:28:09 +01:00