Init

LICENSE

@@ -0,0 +1,10 @@
+Sample Code License
+Version: 1.1
+ 
+IMPORTANT:  This software is supplied to you by École Polytechnique Fédérale de Lausanne (“EPFL”) and Apple Inc. ("Apple") in consideration of your agreement to the following terms, and your use, installation, modification or redistribution of this software constitutes acceptance of these terms.  If you do not agree with these terms, please do not use, install, modify or redistribute this software.
+ 
+In consideration of your agreement to abide by the following terms, and subject to these terms, EPFL and Apple (collectively, “Licensor”) grant you a personal, non-exclusive license, under Licensor’s copyrights in this original software (the " Software"), to use, reproduce, modify and redistribute the Software, with or without modifications, in source and/or binary forms for non-commercial use; provided that if you redistribute the Software in its entirety and without modifications, you must retain this notice and the following text and disclaimers in all such redistributions of the Software. Neither the name, trademarks, service marks or logos of Licensor may be used to endorse or promote products derived from the Software without specific prior written permission from Licensor.  Except as expressly stated in this notice, no other rights or licenses, express or implied, are granted by Licensor herein, including but not limited to any patent rights that may be infringed by your derivative works or by other works in which the Software may be incorporated.
+ 
+The Software is provided by Licensor on an "AS IS" basis.  LICENSOR MAKES NO WARRANTIES, EXPRESS OR IMPLIED, INCLUDING WITHOUT LIMITATION THE IMPLIED WARRANTIES OF NON-INFRINGEMENT, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE, REGARDING THE SOFTWARE OR ITS USE AND OPERATION ALONE OR IN COMBINATION WITH YOUR PRODUCTS. IN NO EVENT SHALL LICENSOR BE LIABLE FOR ANY SPECIAL, INDIRECT, INCIDENTAL OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) ARISING IN ANY WAY OUT OF THE USE, REPRODUCTION, MODIFICATION AND/OR DISTRIBUTION OF THE SOFTWARE, HOWEVER CAUSED AND WHETHER UNDER THEORY OF CONTRACT, TORT (INCLUDING NEGLIGENCE), STRICT LIABILITY OR OTHERWISE, EVEN IF LICENSOR HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ 
+Copyright (C) 2024. All Rights Reserved.

README.md

@@ -0,0 +1,49 @@
+---
+license: other
+license_name: sample-code-license
+license_link: LICENSE
+library_name: ml-4m
+---
+
+# 4M: Massively Multimodal Masked Modeling
+
+*David Mizrahi\*, Roman Bachmann\*, Oguzhan Fatih Kar, Teresa Yeo, Mingfei Gao, Afshin Dehghan, Amir Zamir*
+
+Official implementation and pre-trained models for "4M: Massively Multimodal Masked Modeling" (NeurIPS 2023).
+
+[`Website`](https://4m.epfl.ch) | [`Paper`](https://arxiv.org/abs/2312.06647) | [`GitHub`](https://github.com/apple/ml-4m)
+
+4M is a framework for training "any-to-any" foundation models, using tokenization and masking to scale to many diverse modalities. 
+Models trained using 4M can perform a wide range of vision tasks, transfer well to unseen tasks and modalities, and are flexible and steerable multimodal generative models.
+
+
+## Installation
+For install instructions, please see https://github.com/apple/ml-4m. 
+
+
+## Usage
+
+The RGB tokenizer can be loaded from Hugging Face Hub as follows:
+```python
+from fourm.vq.vqvae import DiVAE
+tok_rgb = DiVAE.from_pretrained('EPFL-VILAB/4M_tokenizers_rgb_16k_224-448')
+```
+
+Please see https://github.com/apple/ml-4m/README_TOKENIZATION.md for more detailed instructions and https://github.com/apple/ml-4m for other tokenizer and 4M model checkpoints.
+
+
+## Citation
+
+If you find this repository helpful, please consider citing our work:
+```
+@inproceedings{mizrahi2023fourm,
+    title={{4M}: Massively Multimodal Masked Modeling},
+    author={David Mizrahi and Roman Bachmann and Oguzhan Fatih Kar and Teresa Yeo and Mingfei Gao and Afshin Dehghan and Amir Zamir},
+    booktitle={Thirty-seventh Conference on Neural Information Processing Systems},
+    year={2023},
+}
+```
+
+## License
+
+The model weights in this repository are released under the Sample Code license as found in the [LICENSE](LICENSE) file.

config.json

@@ -0,0 +1 @@
+{"image_size": 448, "image_size_enc": null, "n_channels": 3, "n_labels": null, "enc_type": "vit_b_enc", "patch_proj": true, "post_mlp": true, "patch_size": 16, "quant_type": "lucid", "codebook_size": 16384, "num_codebooks": 1, "latent_dim": 32, "norm_codes": true, "norm_latents": false, "sync_codebook": false, "undo_std": false, "dec_type": "unet_patched", "num_train_timesteps": 1000, "beta_schedule": "linear", "prediction_type": "sample", "clip_sample": false, "thresholding": true, "conditioning": "add", "zero_terminal_snr": true, "image_size_dec": null, "scheduler": "ddim"}

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7217f035f4ec6f0450b11ca4ee365987e68d3bf52be9bdd252a86a9f3cd3105f
+size 1151785722