ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-30 03:01:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Did some work on SD rescaler. Need to run a long test on it eventually.

Browse Source
This commit is contained in:
Jaret Burkett
2023-08-02 07:59:27 -06:00
parent 2bf3e529ce
commit 1a25b275c8
4 changed files with 96 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/config_modules.py
View File

@@ -42,6 +42,8 @@ class NetworkConfig:
self.linear: int = linear
self.conv: int = kwargs.get('conv', None)
self.alpha: float = kwargs.get('alpha', 1.0)
self.linear_alpha: float = kwargs.get('linear_alpha', self.alpha)
self.conv_alpha: float = kwargs.get('conv_alpha', self.conv)
class TrainConfig: