ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-29 18:51:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Added ability to just erase or enhance concepts from a model

Browse Source
This commit is contained in:
Jaret Burkett
2023-07-24 17:33:45 -06:00
parent 61dd818608
commit 7032717294
2 changed files with 68 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
toolkit/config_modules.py
View File

@@ -65,8 +65,8 @@ class ModelConfig:
class SliderTargetConfig:
def __init__(self, **kwargs):
self.target_class: str = kwargs.get('target_class', '')
self.positive: str = kwargs.get('positive', None)
self.negative: str = kwargs.get('negative', None)
self.positive: str = kwargs.get('positive', '')
self.negative: str = kwargs.get('negative', '')
self.multiplier: float = kwargs.get('multiplier', 1.0)
self.weight: float = kwargs.get('weight', 1.0)