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:
Jaret Burkett
2025-04-13 13:49:13 -06:00
parent 6fb44db6a0
commit ca3ce0f34c
5 changed files with 67 additions and 67 deletions

View File

@@ -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