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

Dataset uploads working

Browse Source
This commit is contained in:
Jaret Burkett
2025-02-20 12:47:01 -07:00
parent b5252b5028
commit 9f6030620f
9 changed files with 565 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
toolkit/lora_special.py
View File

@@ -9,6 +9,7 @@ from typing import List, Optional, Dict, Type, Union
import torch
from diffusers import UNet2DConditionModel, PixArtTransformer2DModel, AuraFlowTransformer2DModel
from transformers import CLIPTextModel
from toolkit.models.lokr import LokrModule
from .config_modules import NetworkConfig
from .lorm import count_parameters