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

Added ability to add models to finetune as plugins. Also added flux2 new arch via that method.

Browse Source
This commit is contained in:
Jaret Burkett
2025-03-27 16:07:00 -06:00
parent e9e30104d3
commit 5365200da1
12 changed files with 936 additions and 1058 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
toolkit/models/cogview4.py
View File

@@ -60,6 +60,7 @@ scheduler_config = {
class CogView4(BaseModel):
arch = 'cogview4'
def __init__(
self,
device,