diff --git a/LICENSE b/LICENSE index bb7f0310..4125a2d2 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 5cad31a0..153e056f 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 c731e4f3..c18728ff 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 5cad31a0..153e056f 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 d8933e22..aa59f919 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 5c34de0c..f0afc261 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 a6989ed0..70554ba5 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 404bc282..22ccb96c 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 737ded30..dc8d2d7e 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 e3b74295..966ac912 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 9e07ce9f..0aa8519b 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 65965896..57fb8b53 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 4c30ac56..49d7b513 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 e07bdd84..6fcf5952 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 e4f294e0..7eef4e28 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 0d0b47eb..e9cc2c1b 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 9a4af64f..41d719ae 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 eb8b97dd..c1a31210 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 9d2bd6bc..1b9c8de6 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 a92d6687..9ed3e6e2 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 66d6fdec..b741c220 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 26202b8c..947543f2 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 c6478ee6..f94003fd 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 16ff7230..58165d70 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 fd27b2e7..e81f744c 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 79acc1e4..880453a0 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 b732e461..44f99edf 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 e27cbad6..302e32ae 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 a7f8c71c..549158a0 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 ed44c6fd..c4b6326d 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 1dba01e3..2d0ae578 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 befcf464..5aee6043 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 51e6df00..5e345508 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 b02665c2..0ffcf302 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 8087bfb9..ef102a2f 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 b1a35dfa..ec60c6ec 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