diff --git a/LICENSE b/LICENSE index bb7f031..4125a2d 100644 --- a/LICENSE +++ b/LICENSE @@ -1,6 +1,6 @@ MIT License -Copyright (c) 2023 ml-explore +Copyright © 2023 Apple Inc. All rights reserved Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal diff --git a/llama/convert.py b/llama/convert.py index 5cad31a..153e056 100644 --- a/llama/convert.py +++ b/llama/convert.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import argparse from itertools import starmap diff --git a/llama/llama.py b/llama/llama.py index c731e4f..c18728f 100644 --- a/llama/llama.py +++ b/llama/llama.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import argparse import math import numpy as np diff --git a/lora/convert.py b/lora/convert.py index 5cad31a..153e056 100644 --- a/lora/convert.py +++ b/lora/convert.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import argparse from itertools import starmap diff --git a/lora/llama.py b/lora/llama.py index d8933e2..aa59f91 100644 --- a/lora/llama.py +++ b/lora/llama.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math import mlx.core as mx diff --git a/lora/lora.py b/lora/lora.py index 5c34de0..f0afc26 100644 --- a/lora/lora.py +++ b/lora/lora.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import argparse import math import numpy as np diff --git a/lora/wikisql.py b/lora/wikisql.py index a6989ed..70554ba 100644 --- a/lora/wikisql.py +++ b/lora/wikisql.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + """ Code to preprocess the WikiSQL dataset adapted from https://github.com/salesforce/WikiSQL and diff --git a/mnist/jax_main.py b/mnist/jax_main.py index 404bc28..22ccb96 100644 --- a/mnist/jax_main.py +++ b/mnist/jax_main.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import jax import jax.numpy as jnp import functools diff --git a/mnist/main.py b/mnist/main.py index 737ded3..dc8d2d7 100644 --- a/mnist/main.py +++ b/mnist/main.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import argparse import time diff --git a/mnist/mnist.py b/mnist/mnist.py index e3b7429..966ac91 100644 --- a/mnist/mnist.py +++ b/mnist/mnist.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import gzip import numpy as np import os diff --git a/mnist/torch_main.py b/mnist/torch_main.py index 9e07ce9..0aa8519 100644 --- a/mnist/torch_main.py +++ b/mnist/torch_main.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import argparse import torch import time diff --git a/stable_diffusion/stable_diffusion/__init__.py b/stable_diffusion/stable_diffusion/__init__.py index 6596589..57fb8b5 100644 --- a/stable_diffusion/stable_diffusion/__init__.py +++ b/stable_diffusion/stable_diffusion/__init__.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import time from typing import Tuple diff --git a/stable_diffusion/stable_diffusion/clip.py b/stable_diffusion/stable_diffusion/clip.py index 4c30ac5..49d7b51 100644 --- a/stable_diffusion/stable_diffusion/clip.py +++ b/stable_diffusion/stable_diffusion/clip.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import mlx.core as mx import mlx.nn as nn diff --git a/stable_diffusion/stable_diffusion/config.py b/stable_diffusion/stable_diffusion/config.py index e07bdd8..6fcf595 100644 --- a/stable_diffusion/stable_diffusion/config.py +++ b/stable_diffusion/stable_diffusion/config.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + from dataclasses import dataclass from typing import Optional, Tuple diff --git a/stable_diffusion/stable_diffusion/model_io.py b/stable_diffusion/stable_diffusion/model_io.py index e4f294e..7eef4e2 100644 --- a/stable_diffusion/stable_diffusion/model_io.py +++ b/stable_diffusion/stable_diffusion/model_io.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import json from functools import partial diff --git a/stable_diffusion/stable_diffusion/sampler.py b/stable_diffusion/stable_diffusion/sampler.py index 0d0b47e..e9cc2c1 100644 --- a/stable_diffusion/stable_diffusion/sampler.py +++ b/stable_diffusion/stable_diffusion/sampler.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + from .config import DiffusionConfig import mlx.core as mx diff --git a/stable_diffusion/stable_diffusion/tokenizer.py b/stable_diffusion/stable_diffusion/tokenizer.py index 9a4af64..41d719a 100644 --- a/stable_diffusion/stable_diffusion/tokenizer.py +++ b/stable_diffusion/stable_diffusion/tokenizer.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import regex diff --git a/stable_diffusion/stable_diffusion/unet.py b/stable_diffusion/stable_diffusion/unet.py index eb8b97d..c1a3121 100644 --- a/stable_diffusion/stable_diffusion/unet.py +++ b/stable_diffusion/stable_diffusion/unet.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math from typing import Optional diff --git a/stable_diffusion/stable_diffusion/vae.py b/stable_diffusion/stable_diffusion/vae.py index 9d2bd6b..1b9c8de 100644 --- a/stable_diffusion/stable_diffusion/vae.py +++ b/stable_diffusion/stable_diffusion/vae.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math from typing import List diff --git a/stable_diffusion/txt2image.py b/stable_diffusion/txt2image.py index a92d668..9ed3e6e 100644 --- a/stable_diffusion/txt2image.py +++ b/stable_diffusion/txt2image.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import argparse from PIL import Image from tqdm import tqdm diff --git a/transformer_lm/datasets.py b/transformer_lm/datasets.py index 66d6fde..b741c22 100644 --- a/transformer_lm/datasets.py +++ b/transformer_lm/datasets.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import io import itertools import numpy as np diff --git a/transformer_lm/jax_main.py b/transformer_lm/jax_main.py index 26202b8..947543f 100644 --- a/transformer_lm/jax_main.py +++ b/transformer_lm/jax_main.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import functools import jax import jax.numpy as jnp diff --git a/transformer_lm/main.py b/transformer_lm/main.py index c6478ee..f94003f 100644 --- a/transformer_lm/main.py +++ b/transformer_lm/main.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math import time diff --git a/transformer_lm/tf_main.py b/transformer_lm/tf_main.py index 16ff723..58165d7 100644 --- a/transformer_lm/tf_main.py +++ b/transformer_lm/tf_main.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math import time diff --git a/transformer_lm/torch_main.py b/transformer_lm/torch_main.py index fd27b2e..e81f744 100644 --- a/transformer_lm/torch_main.py +++ b/transformer_lm/torch_main.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import math import time diff --git a/whisper/benchmark.py b/whisper/benchmark.py index 79acc1e..880453a 100644 --- a/whisper/benchmark.py +++ b/whisper/benchmark.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import time import mlx.core as mx diff --git a/whisper/test.py b/whisper/test.py index b732e46..44f99ed 100644 --- a/whisper/test.py +++ b/whisper/test.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import unittest import mlx.core as mx diff --git a/whisper/whisper/__init__.py b/whisper/whisper/__init__.py index e27cbad..302e32a 100644 --- a/whisper/whisper/__init__.py +++ b/whisper/whisper/__init__.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + from . import load_models from . import audio from . import decoding diff --git a/whisper/whisper/audio.py b/whisper/whisper/audio.py index a7f8c71..549158a 100644 --- a/whisper/whisper/audio.py +++ b/whisper/whisper/audio.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import os from functools import lru_cache from subprocess import CalledProcessError, run diff --git a/whisper/whisper/decoding.py b/whisper/whisper/decoding.py index ed44c6f..c4b6326 100644 --- a/whisper/whisper/decoding.py +++ b/whisper/whisper/decoding.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + from dataclasses import dataclass, field, replace from typing import Dict, Iterable, List, Optional, Sequence, Tuple, Union import mlx.core as mx diff --git a/whisper/whisper/load_models.py b/whisper/whisper/load_models.py index 1dba01e..2d0ae57 100644 --- a/whisper/whisper/load_models.py +++ b/whisper/whisper/load_models.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import hashlib import os import urllib diff --git a/whisper/whisper/timing.py b/whisper/whisper/timing.py index befcf46..5aee604 100644 --- a/whisper/whisper/timing.py +++ b/whisper/whisper/timing.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import itertools import subprocess import warnings diff --git a/whisper/whisper/tokenizer.py b/whisper/whisper/tokenizer.py index 51e6df0..5e34550 100644 --- a/whisper/whisper/tokenizer.py +++ b/whisper/whisper/tokenizer.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import base64 import os import string diff --git a/whisper/whisper/torch_whisper.py b/whisper/whisper/torch_whisper.py index b02665c..0ffcf30 100644 --- a/whisper/whisper/torch_whisper.py +++ b/whisper/whisper/torch_whisper.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import base64 import gzip from dataclasses import dataclass diff --git a/whisper/whisper/transcribe.py b/whisper/whisper/transcribe.py index 8087bfb..ef102a2 100644 --- a/whisper/whisper/transcribe.py +++ b/whisper/whisper/transcribe.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import mlx.core as mx import numpy as np import sys diff --git a/whisper/whisper/whisper.py b/whisper/whisper/whisper.py index b1a35df..ec60c6e 100644 --- a/whisper/whisper/whisper.py +++ b/whisper/whisper/whisper.py @@ -1,3 +1,5 @@ +# Copyright © 2023 Apple Inc. + import base64 import gzip from dataclasses import dataclass