Add the notice

flux/dreambooth.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 import argparse
 import json
 import time

flux/flux/__init__.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 import math
 import time
 from typing import Tuple

flux/flux/autoencoder.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 from dataclasses import dataclass
 from typing import List
 

flux/flux/clip.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 from dataclasses import dataclass
 from typing import List, Optional
 

flux/flux/layers.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 import math
 from dataclasses import dataclass
 from functools import partial

flux/flux/lora.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 import math
 
 import mlx.core as mx

flux/flux/model.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 from dataclasses import dataclass
 from typing import Optional
 

flux/flux/sampler.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 import math
 from functools import lru_cache
 

flux/flux/t5.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 import math
 from dataclasses import dataclass
 from typing import List, Optional, Tuple

flux/flux/tokenizers.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 import mlx.core as mx
 import regex
 from sentencepiece import SentencePieceProcessor

flux/flux/utils.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 import json
 import os
 from dataclasses import dataclass

flux/txt2image.py

@@ -1,3 +1,5 @@
+# Copyright © 2024 Apple Inc.
+
 import argparse
 
 import mlx.core as mx