ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

t2i training working from what I can tell at least

Browse Source
This commit is contained in:
Jaret Burkett
2023-09-17 15:56:43 -06:00
parent 181f237a7b
commit 61badf85a7
5 changed files with 214 additions and 174 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/saving.py
View File

@@ -184,7 +184,7 @@ def save_t2i_from_diffusers(
def load_t2i_model(
path_to_file,
device: Union[str, torch.device] = 'cpu',
device: Union[str] = 'cpu',
dtype: torch.dtype = torch.float32
):
raw_state_dict = load_file(path_to_file, device)