mirror of
https://github.com/ostris/ai-toolkit.git
synced 2026-03-13 22:49:48 +00:00
Make it easier to designate lora blocks for new models. Improve i2v adapter speed. Fix issue with i2v adapter where cached torch tensor was wrong range.
This commit is contained in:
@@ -4,7 +4,7 @@ import json
|
||||
import random
|
||||
import shutil
|
||||
import typing
|
||||
from typing import Union, List, Literal, Iterator
|
||||
from typing import Optional, Union, List, Literal, Iterator
|
||||
import sys
|
||||
import os
|
||||
from collections import OrderedDict
|
||||
@@ -3079,3 +3079,7 @@ class StableDiffusion:
|
||||
def condition_noisy_latents(self, latents: torch.Tensor, batch:'DataLoaderBatchDTO'):
|
||||
# can be overridden in child classes to condition latents before noise prediction
|
||||
return latents
|
||||
|
||||
def get_transformer_block_names(self) -> Optional[List[str]]:
|
||||
# override in child classes to get transformer block names for lora targeting
|
||||
return None
|
||||
|
||||
Reference in New Issue
Block a user