Upload folder using huggingface_hub

README.md

@@ -0,0 +1,67 @@
+---
+license: mit
+license_name: deepseek
+license_link: LICENSE
+pipeline_tag: any-to-any
+library_name: transformers
+tags:
+- muiltimodal
+- text-to-image
+- unified-model
+---
+
+## 0. Update
+**2024.10.20**: We have uploaded the correct `tokenizer_config.json`. The previous file was missing the `pad_token`, which caused poor visual generation results.
+
+
+## 1. Introduction
+
+Janus is a novel autoregressive framework that unifies multimodal understanding and generation. 
+It addresses the limitations of previous approaches by decoupling visual encoding into separate pathways, while still utilizing a single, unified transformer architecture for processing. The decoupling not only alleviates the conflict between the visual encoder’s roles in understanding and generation, but also enhances the framework’s flexibility. 
+Janus surpasses previous unified model and matches or exceeds the performance of task-specific models. 
+The simplicity, high flexibility, and effectiveness of Janus make it a strong candidate for next-generation unified multimodal models.
+
+[Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generation](https://arxiv.org/abs/2410.13848)
+
+[**Github Repository**](https://github.com/deepseek-ai/Janus)
+
+<div align="center">
+<img alt="image" src="teaser.png" style="width:90%;">
+</div>
+
+
+### 2. Model Summary
+
+Janus is a unified understanding and generation MLLM, which decouples visual encoding for multimodal understanding and generation. 
+Janus is constructed based on the DeepSeek-LLM-1.3b-base which is trained on an approximate corpus of 500B text tokens.
+For multimodal understanding, it uses the [SigLIP-L](https://huggingface.co/timm/ViT-L-16-SigLIP-384) as the vision encoder, which supports 384 x 384 image input. For image generation, Janus uses the tokenizer from [here](https://github.com/FoundationVision/LlamaGen) with a downsample rate of 16.
+
+<div align="center">
+<img alt="image" src="arch.jpg" style="width:90%;">
+</div>
+
+## 3. Quick Start
+
+Please refer to [**Github Repository**](https://github.com/deepseek-ai/Janus)
+
+
+## 4. License
+
+This code repository is licensed under [the MIT License](https://github.com/deepseek-ai/DeepSeek-LLM/blob/HEAD/LICENSE-CODE). The use of Janus models is subject to [DeepSeek Model License](https://github.com/deepseek-ai/DeepSeek-LLM/blob/HEAD/LICENSE-MODEL).
+## 5. Citation
+
+```
+@misc{wu2024janus,
+      title={Janus: Decoupling Visual Encoding for Unified Multimodal Understanding and Generation}, 
+      author={Chengyue Wu and Xiaokang Chen and Zhiyu Wu and Yiyang Ma and Xingchao Liu and Zizheng Pan and Wen Liu and Zhenda Xie and Xingkai Yu and Chong Ruan and Ping Luo},
+      year={2024},
+      eprint={2410.13848},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/2410.13848}, 
+}
+```
+
+## 6. Contact
+
+If you have any questions, please raise an issue or contact us at [service@deepseek.com](mailto:service@deepseek.com).

config.json

@@ -0,0 +1,17 @@
+{
+  "architectures": [
+    "LlamaForCausalLM"
+  ],
+  "hidden_size": 2048,
+  "intermediate_size": 5632,
+  "max_position_embeddings": 16384,
+  "model_type": "llama",
+  "num_attention_heads": 16,
+  "num_hidden_layers": 24,
+  "num_key_value_heads": 16,
+  "torch_dtype": "bfloat16",
+  "vocab_size": 102400,
+  "eos_token_id": 100001,
+  "pad_token_id": 100000,
+  "transformers_version": "4.38.2"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9328d29bb8feca7334b9fdcdcc644d6176ad4b6873117e6225fce64c9e3dda53
+size 3305337584

preprocessor_config.json

@@ -0,0 +1,23 @@
+{
+  "background_color": [
+    127,
+    127,
+    127
+  ],
+  "do_normalize": true,
+  "image_mean": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "image_processor_type": "VLMImageProcessor",
+  "image_size": 384,
+  "image_std": [
+    0.5,
+    0.5,
+    0.5
+  ],
+  "min_size": 14,
+  "processor_class": "VLChatProcessor",
+  "rescale_factor": 0.00392156862745098
+}

processor_config.json

@@ -0,0 +1,9 @@
+{
+  "add_special_token": false,
+  "ignore_id": -100,
+  "image_tag": "<image_placeholder>",
+  "mask_prompt": true,
+  "num_image_tokens": 576,
+  "processor_class": "VLChatProcessor",
+  "sft_format": "deepseek_old"
+}

rename_key.py

@@ -0,0 +1,13 @@
+from safetensors import safe_open
+
+from safetensors.torch import save_file
+
+
+if __name__ == "__main__":
+    tensors = {}
+    with safe_open("model.safetensors", framework="pt", device=0) as f:
+        for k in f.keys():
+            if k.startswith("language_model."):
+                tensors[k.split("language_model.")[1]] = f.get_tensor(k)
+
+    save_file(tensors, "model_fix.safetensors")

special_tokens_map.json

@@ -0,0 +1,5 @@
+{
+  "bos_token": "<｜begin▁of▁sentence｜>",
+  "eos_token": "<｜end▁of▁sentence｜>",
+  "pad_token": "<｜▁pad▁｜>"
+}