From 94aba5892a9541031c98eb4bbafd8afde6ba49b7 Mon Sep 17 00:00:00 2001 From: Brandon Rising Date: Wed, 21 Aug 2024 15:53:58 -0400 Subject: [PATCH] Attribute black-forest-labs/flux for much of the flux code --- invokeai/backend/flux/math.py | 2 ++ invokeai/backend/flux/model.py | 2 ++ invokeai/backend/flux/modules/autoencoder.py | 2 ++ invokeai/backend/flux/modules/conditioner.py | 2 ++ invokeai/backend/flux/modules/layers.py | 2 ++ invokeai/backend/flux/sampling.py | 2 ++ 6 files changed, 12 insertions(+) diff --git a/invokeai/backend/flux/math.py b/invokeai/backend/flux/math.py index 0156bb6a20..aa719b7c07 100644 --- a/invokeai/backend/flux/math.py +++ b/invokeai/backend/flux/math.py @@ -1,3 +1,5 @@ +# Initially pulled from https://github.com/black-forest-labs/flux + import torch from einops import rearrange from torch import Tensor diff --git a/invokeai/backend/flux/model.py b/invokeai/backend/flux/model.py index f7ef25bf4f..5358ddf0bc 100644 --- a/invokeai/backend/flux/model.py +++ b/invokeai/backend/flux/model.py @@ -1,3 +1,5 @@ +# Initially pulled from https://github.com/black-forest-labs/flux + from dataclasses import dataclass import torch diff --git a/invokeai/backend/flux/modules/autoencoder.py b/invokeai/backend/flux/modules/autoencoder.py index 75159f711f..ae003261e7 100644 --- a/invokeai/backend/flux/modules/autoencoder.py +++ b/invokeai/backend/flux/modules/autoencoder.py @@ -1,3 +1,5 @@ +# Initially pulled from https://github.com/black-forest-labs/flux + from dataclasses import dataclass import torch diff --git a/invokeai/backend/flux/modules/conditioner.py b/invokeai/backend/flux/modules/conditioner.py index 974ad64ab3..de6d8256c4 100644 --- a/invokeai/backend/flux/modules/conditioner.py +++ b/invokeai/backend/flux/modules/conditioner.py @@ -1,3 +1,5 @@ +# Initially pulled from https://github.com/black-forest-labs/flux + from torch import Tensor, nn from transformers import PreTrainedModel, PreTrainedTokenizer diff --git a/invokeai/backend/flux/modules/layers.py b/invokeai/backend/flux/modules/layers.py index d93dddba0f..23dc2448d3 100644 --- a/invokeai/backend/flux/modules/layers.py +++ b/invokeai/backend/flux/modules/layers.py @@ -1,3 +1,5 @@ +# Initially pulled from https://github.com/black-forest-labs/flux + import math from dataclasses import dataclass diff --git a/invokeai/backend/flux/sampling.py b/invokeai/backend/flux/sampling.py index 675728a94b..9917d63a8b 100644 --- a/invokeai/backend/flux/sampling.py +++ b/invokeai/backend/flux/sampling.py @@ -1,3 +1,5 @@ +# Initially pulled from https://github.com/black-forest-labs/flux + import math from typing import Callable