Attribute black-forest-labs/flux for much of the flux code

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

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

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

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
 

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
 

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