Fix styling/lint

This commit is contained in:
Brandon Rising
2024-08-20 13:05:31 -04:00
parent da766f5a7e
commit dd24f83d43
12 changed files with 27 additions and 788 deletions

View File

@ -5,7 +5,7 @@ import torch
from einops import rearrange
from torch import Tensor, nn
from ..math import attention, rope
from invokeai.backend.flux.math import attention, rope
class EmbedND(nn.Module):

View File

@ -6,8 +6,8 @@ from einops import rearrange, repeat
from torch import Tensor
from tqdm import tqdm
from .model import Flux
from .modules.conditioner import HFEncoder
from invokeai.backend.flux.model import Flux
from invokeai.backend.flux.modules.conditioner import HFEncoder
def get_noise(