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

Huge memory optimizations, many big fixes

Browse Source
This commit is contained in:
Jaret Burkett
2023-08-27 17:48:02 -06:00
parent cc49786ee9
commit c446f768ea
15 changed files with 86 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jobs/process/GenerateProcess.py
View File

@@ -16,9 +16,9 @@ import random
class GenerateConfig:
prompts: List[str]
def __init__(self, **kwargs):
self.prompts: List[str]
self.sampler = kwargs.get('sampler', 'ddpm')
self.width = kwargs.get('width', 512)
self.height = kwargs.get('height', 512)