Upload SDNQ Uint Static quantized Lens-Turbo

.gitattributes

@@ -33,3 +33,24 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p01_original.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p01_sdnq_uint_static.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p02_original.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p02_sdnq_uint_static.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p03_original.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p03_sdnq_uint_static.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p04_original.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p04_sdnq_uint_static.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p05_original.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p05_sdnq_uint_static.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p06_original.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p06_sdnq_uint_static.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p07_original.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p07_sdnq_uint_static.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p08_original.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p08_sdnq_uint_static.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p09_original.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p09_sdnq_uint_static.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p10_original.png filter=lfs diff=lfs merge=lfs -text
+assets/comparison/p10_sdnq_uint_static.png filter=lfs diff=lfs merge=lfs -text
+tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,199 @@
+---
+license: mit
+language:
+- en
+- ru
+pipeline_tag: text-to-image
+tags:
+- diffusers
+- safetensors
+- text-to-image
+- LensPipeline
+- sdnq
+- quantized
+- uint4
+- static-quantization
+base_model: microsoft/Lens-Turbo
+---
+
+# Lens-Turbo SDNQ Uint Static
+
+This is an SDNQ static UINT4 quantized variant of [microsoft/Lens-Turbo](https://huggingface.co/microsoft/Lens-Turbo).
+It keeps the Lens pipeline structure intact and focuses quantization on the denoising transformer, which is the main generation component.
+
+The original Lens card describes Lens as a 3.8B-parameter text-to-image model using dense-caption training, GPT-OSS multi-layer text features, and the FLUX.2 semantic VAE. Lens-Turbo is the distilled 4-step variant.
+
+## Quantization
+
+| Field | Value |
+| --- | --- |
+| Method | SDNQ Uint Static |
+| Quantized component | `transformer` / `LensTransformer2DModel` |
+| Weight dtype | `uint4` |
+| Quantized matmul | enabled |
+| Quantized matmul dtype | `int8` |
+| Static quantization | enabled |
+| Dynamic quantization | disabled |
+| SVDQuant | disabled |
+| Hadamard rotation | disabled |
+| Convolution quantization | disabled |
+| Embedding quantization | disabled |
+| Text encoder | unchanged from source checkpoint |
+| VAE | unchanged from source checkpoint |
+| Compute dtype | `torch.bfloat16` |
+
+Raw config:
+
+```json
+{
+  "weights_dtype": "uint4",
+  "quantized_matmul_dtype": "int8",
+  "group_size": "0",
+  "use_static_quantization": "True",
+  "use_dynamic_quantization": "False",
+  "use_quantized_matmul": "True",
+  "use_svd": "False",
+  "use_hadamard": "False",
+  "quant_conv": "False",
+  "quant_embedding": "False",
+  "dequantize_fp32": "False",
+  "quantization_device": "cuda",
+  "return_device": "cuda"
+}
+```
+
+## Usage
+
+Install the Lens inference code and SDNQ, then download the repo snapshot and load the quantized transformer explicitly:
+
+```python
+import torch
+from huggingface_hub import snapshot_download
+from lens import LensPipeline, LensTransformer2DModel
+from sdnq import load_sdnq_model
+
+repo_id = "WaveCut/Lens-Turbo-SDNQ-Uint-Static"
+model_dir = snapshot_download(repo_id)
+
+transformer = load_sdnq_model(
+    model_dir + "/transformer",
+    model_cls=LensTransformer2DModel,
+    dtype=torch.bfloat16,
+    device=torch.device("cuda"),
+    dequantize_fp32=False,
+    use_quantized_matmul=True,
+)
+
+pipe = LensPipeline.from_pretrained(
+    model_dir,
+    transformer=transformer,
+    torch_dtype=torch.bfloat16,
+).to("cuda")
+
+image = pipe(
+    "A cat holding a sign that says hello world",
+    base_resolution=1024,
+    aspect_ratio="1:1",
+    num_inference_steps=4,
+    guidance_scale=1.0,
+    generator=torch.Generator("cuda").manual_seed(0),
+).images[0]
+```
+
+## Benchmark
+
+Benchmark date: 2026-05-24. Hardware: RunPod NVIDIA H100 80GB HBM3, PyTorch 2.8.0 CUDA 12.8 container, no network volume, local container disk only.
+Each prompt used `base_resolution=1024`, `aspect_ratio="1:1"`, `num_inference_steps=4`, `guidance_scale=1.0`, `torch.bfloat16`, and a fixed CUDA seed.
+
+| Metric | Original Lens-Turbo | SDNQ Uint Static |
+| --- | ---: | ---: |
+| Load time, seconds | 59.316 | 16.861 |
+| Load peak allocated VRAM, GB | 20.807 | 15.281 |
+| Load peak reserved VRAM, GB | 20.928 | 15.344 |
+| Load end allocated VRAM, GB | 20.807 | 15.281 |
+| Average prompt runtime, seconds | 2.206 | 3.394 |
+| Average runtime delta | baseline | +53.8% |
+
+## 10-Prompt Comparison Matrix
+
+| ID | Scenario | Seed | Original time, s | Quant time, s | Delta | Original peak reserved VRAM, GB | Quant peak reserved VRAM, GB | Original | Quant |
+| --- | --- | ---: | ---: | ---: | ---: | ---: | ---: | --- | --- |
+| P01 | Orbital Night Market | 101 | 6.902 | 7.980 | +15.6% | 25.438 | 19.746 | ![](assets/comparison/p01_original.png) | ![](assets/comparison/p01_sdnq_uint_static.png) |
+| P02 | Arctic Research Desk | 102 | 1.347 | 3.126 | +132.1% | 25.434 | 19.740 | ![](assets/comparison/p02_original.png) | ![](assets/comparison/p02_sdnq_uint_static.png) |
+| P03 | Victorian Automaton Repair | 103 | 2.901 | 2.846 | -1.9% | 25.438 | 19.744 | ![](assets/comparison/p03_original.png) | ![](assets/comparison/p03_sdnq_uint_static.png) |
+| P04 | Mars Greenhouse Control Room | 104 | 1.186 | 2.846 | +140.0% | 25.438 | 19.725 | ![](assets/comparison/p04_original.png) | ![](assets/comparison/p04_sdnq_uint_static.png) |
+| P05 | Lost Railway Poster Wall | 105 | 1.185 | 2.877 | +142.8% | 25.438 | 19.725 | ![](assets/comparison/p05_original.png) | ![](assets/comparison/p05_sdnq_uint_static.png) |
+| P06 | Miniature Courtroom Diorama | 106 | 1.186 | 2.836 | +139.1% | 25.438 | 19.744 | ![](assets/comparison/p06_original.png) | ![](assets/comparison/p06_sdnq_uint_static.png) |
+| P07 | Rainy Seoul Book Cafe | 107 | 1.243 | 2.835 | +128.1% | 25.438 | 19.744 | ![](assets/comparison/p07_original.png) | ![](assets/comparison/p07_sdnq_uint_static.png) |
+| P08 | Oceanographic Expedition Map | 108 | 1.181 | 2.847 | +141.1% | 25.438 | 19.744 | ![](assets/comparison/p08_original.png) | ![](assets/comparison/p08_sdnq_uint_static.png) |
+| P09 | Renaissance Lab Notebook | 109 | 1.180 | 2.855 | +141.9% | 25.438 | 19.719 | ![](assets/comparison/p09_original.png) | ![](assets/comparison/p09_sdnq_uint_static.png) |
+| P10 | Russian Provincial Print Shop | 110 | 3.749 | 2.888 | -23.0% | 25.459 | 19.766 | ![](assets/comparison/p10_original.png) | ![](assets/comparison/p10_sdnq_uint_static.png) |
+
+## Full Prompts
+
+<details><summary>P01 - Orbital Night Market</summary>
+
+A dense cinematic night market inside a transparent orbital habitat, with Earth curving below the glass floor, vendors selling glowing algae noodles and tiny repair drones, rain droplets floating in zero gravity, reflections on wet metal, and at least six readable signs in different places: a vertical neon sign saying "ORBITAL TEA HOUSE", a handwritten chalk menu saying "NO GRAVITY REFUNDS", a yellow safety placard saying "MAG BOOTS REQUIRED", a small receipt printer label saying "BAY 12 PICKUP", a red banner saying "FRESH SYNTH-MANGO", and a blue customs notice saying "DECLARE ALL MOON ROCKS". Ultra detailed, wide angle, layered crowd, realistic lens flare, crisp small typography.
+
+</details>
+<details><summary>P02 - Arctic Research Desk</summary>
+
+A top-down documentary photo of an Arctic climate research desk inside a weather station during a blizzard, with ice crystals on the window, a rugged laptop displaying a complex map, three paper field notebooks, sample vials, a steaming enamel mug, and long English text on multiple objects: the notebook cover reads "FIELD LOG: STATION NORD, WEEK 17", a whiteboard in the background reads "CORE DEPTH 42.8m / TEMP -31C / WIND 62 km/h", a red tag on a sample tube reads "DO NOT THAW", and a printed memo reads "CALIBRATE SENSORS BEFORE SUNRISE". Natural cold light, precise shadows, photorealistic texture, no blurry text.
+
+</details>
+<details><summary>P03 - Victorian Automaton Repair</summary>
+
+A richly detailed Victorian workshop where a brass clockwork automaton is being repaired under green banker lamps, with tiny gears, pearl inlays, oiled leather belts, smoke from a soldering iron, magnifying glass distortion, and handwritten labels everywhere. The main blueprint title must read "AUTOMATON HAND ASSEMBLY REV. C", a drawer label says "SPRINGS / EYES / MEMORY CAMS", a dangling tag says "CLIENT: LADY ADA", and a note pinned to the wall says "DO NOT WIND PAST MIDNIGHT". Moody chiaroscuro, shallow depth of field, extremely fine mechanical detail.
+
+</details>
+<details><summary>P04 - Mars Greenhouse Control Room</summary>
+
+A believable Mars greenhouse control room at dawn, red dust outside the curved windows, rows of tomatoes and dwarf wheat under violet grow lights, condensation on transparent tubes, a tired botanist reflected in a touchscreen, and several readable UI panels in English: "OXYGEN LOOP STABLE", "WATER RECOVERY 98.4%", "SECTOR C: POLLINATION DRONES ACTIVE", and a sticky note saying "Tell Earth the basil survived". Technical but warm, high resolution, realistic sci-fi, detailed glass and plant textures.
+
+</details>
+<details><summary>P05 - Lost Railway Poster Wall</summary>
+
+An abandoned underground railway platform turned into an accidental archive of travel posters, peeling ceramic tiles, puddles reflecting amber emergency lights, old suitcases, vines growing through cracked concrete, and five large posters with distinct readable titles: "THE NORTHERN COMET EXPRESS", "SLEEPER TO ISTANBUL", "MIDNIGHT PLATFORM 7", "COASTAL ROUTE REOPENING SOON", and "KEEP YOUR TICKET VISIBLE". Cinematic composition, wet surfaces, layered typography, realistic grime, strong perspective down the tracks.
+
+</details>
+<details><summary>P06 - Miniature Courtroom Diorama</summary>
+
+A hyperreal macro photograph of a miniature courtroom diorama built inside an antique wooden drawer, with tiny judge bench, brass lamps, dust motes, paper exhibits smaller than postage stamps, a mouse-sized witness chair, and readable text on tiny documents: a case file labeled "CASE 1842-B: THE MISSING ORRERY", an evidence tag saying "EXHIBIT C", a court calendar reading "HEARING AT 9:30", and a placard on the judge bench saying "TRUTH IN SMALL THINGS". Macro lens, tactile materials, careful scale cues.
+
+</details>
+<details><summary>P07 - Rainy Seoul Book Cafe</summary>
+
+A cozy but complex rainy evening scene in a narrow Seoul book cafe, viewed through a window covered in raindrops, shelves packed with art books, two students annotating a map, a barista steaming milk, warm tungsten light, street reflections, and multiple readable English text elements: a chalkboard says "TONIGHT: QUIET READING CLUB", a receipt says "OAT LATTE / CINNAMON BUN", a book spine says "ARCHITECTURE OF DREAMS", and a window sticker says "OPEN UNTIL THE LAST TRAIN". Photorealistic, cinematic, intricate reflections.
+
+</details>
+<details><summary>P08 - Oceanographic Expedition Map</summary>
+
+A dramatic captain's table aboard a storm-tossed oceanographic research vessel, with a wet nautical chart, brass dividers, sonar printouts, bioluminescent plankton glowing in a glass jar, a cracked tablet, and readable labels distributed across the image: "TRENCH SURVEY LINE B", "DEPTH 10,928m", "ROV SIGNAL WEAK", "SAMPLE: BLUE VENT WATER", and a torn note saying "If the lights pulse twice, turn back". High detail, realistic water droplets, dark blue-green atmosphere, sharp text.
+
+</details>
+<details><summary>P09 - Renaissance Lab Notebook</summary>
+
+An alternate-history Renaissance laboratory where an astronomer-painter is combining oil pigments with early electrical apparatus, with celestial globes, copper coils, stained glass sunlight, anatomical sketches, a half-finished portrait, and Latin-English notebook text visible on several pages: "LIGHT STUDY: BLUE VERDITER", "GALVANIC TEST NO. 8", "VENUS RISES BEFORE DAWN", and a folded letter sealed in wax reading "FOR THE WORKSHOP MASTER ONLY". Painterly realism, ornate detail, coherent objects, readable calligraphy.
+
+</details>
+<details><summary>P10 - Russian Provincial Print Shop</summary>
+
+Сложная фотореалистичная сцена в старой провинциальной типографии поздним зимним вечером: за большим деревянным столом лежат металлические литеры, корректурные листы, линейки, чашка крепкого чая, заснеженное окно, тусклая лампа и следы типографской краски на пальцах наборщика. На разных элементах изображения должен быть длинный и хорошо читаемый русский текст: на вывеске над дверью написано "ТИПОГРАФИЯ СЕВЕРНЫЙ ЛИСТОК", на корректуре заголовок "СРОЧНО В НОМЕР: ГОРОДСКОЙ СОВЕТ ОТКРЫВАЕТ НОВУЮ БИБЛИОТЕКУ", на маленькой записке фраза "Проверить букву Ё во втором абзаце", а на календаре дата "Пятница, 24 января". Много бытовых деталей, глубокие тени, реалистичная кириллица, никакой размытой каши вместо текста.
+
+</details>
+
+## Notes
+
+This checkpoint is intended for research and evaluation. It inherits the upstream Lens limitations and responsible AI considerations from the source model.
+The comparison images are deterministic for the listed seeds under this environment but may vary across driver, PyTorch, SDNQ, or kernel versions.
+
+## Citation
+
+```bibtex
+@article{zhao2026lens,
+  title   = {Lens: Rethinking Training Efficiency for Foundational Text-to-Image Models},
+  author  = {Guo, Baining and Luo, Chong and Chen, Dong and Chen, Dongdong and Wei, Fangyun and Li, Ji and Bao, Jianmin and Zhang, Jiawei and Zhao, Jinjing and Shi, Lei and Yang, Qinhong and Zhang, Sirui and Wu, Xiuyu and Feng, Xuelu and Lu, Yan and Dong, Yanchen and Yue, Yang and Wang, Yitong and Chen, Yunuo and Liang, Zhiyang and Wan, Ziyu},
+  journal = {arXiv preprint arXiv:2605.21573},
+  year    = {2026}
+}
+```

benchmark_metrics.json

@@ -0,0 +1,19 @@
+{
+  "load": {
+    "base": {
+      "load_time_s": 59.316,
+      "peak_allocated_gb": 20.807,
+      "peak_reserved_gb": 20.928,
+      "end_allocated_gb": 20.807,
+      "end_reserved_gb": 20.928
+    },
+    "quant": {
+      "load_time_s": 16.861,
+      "peak_allocated_gb": 15.281,
+      "peak_reserved_gb": 15.344,
+      "end_allocated_gb": 15.281,
+      "end_reserved_gb": 15.344
+    }
+  },
+  "quantization_time_s": null
+}

comparison_matrix.json

@@ -0,0 +1,312 @@
+[
+  {
+    "id": "P01",
+    "title": "Orbital Night Market",
+    "seed": 101,
+    "prompt": "A dense cinematic night market inside a transparent orbital habitat, with Earth curving below the glass floor, vendors selling glowing algae noodles and tiny repair drones, rain droplets floating in zero gravity, reflections on wet metal, and at least six readable signs in different places: a vertical neon sign saying \"ORBITAL TEA HOUSE\", a handwritten chalk menu saying \"NO GRAVITY REFUNDS\", a yellow safety placard saying \"MAG BOOTS REQUIRED\", a small receipt printer label saying \"BAY 12 PICKUP\", a red banner saying \"FRESH SYNTH-MANGO\", and a blue customs notice saying \"DECLARE ALL MOON ROCKS\". Ultra detailed, wide angle, layered crowd, realistic lens flare, crisp small typography.",
+    "base": {
+      "id": "P01",
+      "title": "Orbital Night Market",
+      "seed": 101,
+      "prompt": "A dense cinematic night market inside a transparent orbital habitat, with Earth curving below the glass floor, vendors selling glowing algae noodles and tiny repair drones, rain droplets floating in zero gravity, reflections on wet metal, and at least six readable signs in different places: a vertical neon sign saying \"ORBITAL TEA HOUSE\", a handwritten chalk menu saying \"NO GRAVITY REFUNDS\", a yellow safety placard saying \"MAG BOOTS REQUIRED\", a small receipt printer label saying \"BAY 12 PICKUP\", a red banner saying \"FRESH SYNTH-MANGO\", and a blue customs notice saying \"DECLARE ALL MOON ROCKS\". Ultra detailed, wide angle, layered crowd, realistic lens flare, crisp small typography.",
+      "image": "assets/comparison/p01_original.png",
+      "time_s": 6.902,
+      "peak_allocated_gb": 23.245,
+      "peak_reserved_gb": 25.438,
+      "end_allocated_gb": 20.841,
+      "end_reserved_gb": 25.438
+    },
+    "quant": {
+      "id": "P01",
+      "title": "Orbital Night Market",
+      "seed": 101,
+      "prompt": "A dense cinematic night market inside a transparent orbital habitat, with Earth curving below the glass floor, vendors selling glowing algae noodles and tiny repair drones, rain droplets floating in zero gravity, reflections on wet metal, and at least six readable signs in different places: a vertical neon sign saying \"ORBITAL TEA HOUSE\", a handwritten chalk menu saying \"NO GRAVITY REFUNDS\", a yellow safety placard saying \"MAG BOOTS REQUIRED\", a small receipt printer label saying \"BAY 12 PICKUP\", a red banner saying \"FRESH SYNTH-MANGO\", and a blue customs notice saying \"DECLARE ALL MOON ROCKS\". Ultra detailed, wide angle, layered crowd, realistic lens flare, crisp small typography.",
+      "image": "assets/comparison/p01_sdnq_uint_static.png",
+      "time_s": 7.98,
+      "peak_allocated_gb": 17.687,
+      "peak_reserved_gb": 19.746,
+      "end_allocated_gb": 15.283,
+      "end_reserved_gb": 19.746
+    },
+    "speed_delta_pct": 15.6
+  },
+  {
+    "id": "P02",
+    "title": "Arctic Research Desk",
+    "seed": 102,
+    "prompt": "A top-down documentary photo of an Arctic climate research desk inside a weather station during a blizzard, with ice crystals on the window, a rugged laptop displaying a complex map, three paper field notebooks, sample vials, a steaming enamel mug, and long English text on multiple objects: the notebook cover reads \"FIELD LOG: STATION NORD, WEEK 17\", a whiteboard in the background reads \"CORE DEPTH 42.8m / TEMP -31C / WIND 62 km/h\", a red tag on a sample tube reads \"DO NOT THAW\", and a printed memo reads \"CALIBRATE SENSORS BEFORE SUNRISE\". Natural cold light, precise shadows, photorealistic texture, no blurry text.",
+    "base": {
+      "id": "P02",
+      "title": "Arctic Research Desk",
+      "seed": 102,
+      "prompt": "A top-down documentary photo of an Arctic climate research desk inside a weather station during a blizzard, with ice crystals on the window, a rugged laptop displaying a complex map, three paper field notebooks, sample vials, a steaming enamel mug, and long English text on multiple objects: the notebook cover reads \"FIELD LOG: STATION NORD, WEEK 17\", a whiteboard in the background reads \"CORE DEPTH 42.8m / TEMP -31C / WIND 62 km/h\", a red tag on a sample tube reads \"DO NOT THAW\", and a printed memo reads \"CALIBRATE SENSORS BEFORE SUNRISE\". Natural cold light, precise shadows, photorealistic texture, no blurry text.",
+      "image": "assets/comparison/p02_original.png",
+      "time_s": 1.347,
+      "peak_allocated_gb": 23.245,
+      "peak_reserved_gb": 25.434,
+      "end_allocated_gb": 20.841,
+      "end_reserved_gb": 25.434
+    },
+    "quant": {
+      "id": "P02",
+      "title": "Arctic Research Desk",
+      "seed": 102,
+      "prompt": "A top-down documentary photo of an Arctic climate research desk inside a weather station during a blizzard, with ice crystals on the window, a rugged laptop displaying a complex map, three paper field notebooks, sample vials, a steaming enamel mug, and long English text on multiple objects: the notebook cover reads \"FIELD LOG: STATION NORD, WEEK 17\", a whiteboard in the background reads \"CORE DEPTH 42.8m / TEMP -31C / WIND 62 km/h\", a red tag on a sample tube reads \"DO NOT THAW\", and a printed memo reads \"CALIBRATE SENSORS BEFORE SUNRISE\". Natural cold light, precise shadows, photorealistic texture, no blurry text.",
+      "image": "assets/comparison/p02_sdnq_uint_static.png",
+      "time_s": 3.126,
+      "peak_allocated_gb": 17.687,
+      "peak_reserved_gb": 19.74,
+      "end_allocated_gb": 15.283,
+      "end_reserved_gb": 19.74
+    },
+    "speed_delta_pct": 132.1
+  },
+  {
+    "id": "P03",
+    "title": "Victorian Automaton Repair",
+    "seed": 103,
+    "prompt": "A richly detailed Victorian workshop where a brass clockwork automaton is being repaired under green banker lamps, with tiny gears, pearl inlays, oiled leather belts, smoke from a soldering iron, magnifying glass distortion, and handwritten labels everywhere. The main blueprint title must read \"AUTOMATON HAND ASSEMBLY REV. C\", a drawer label says \"SPRINGS / EYES / MEMORY CAMS\", a dangling tag says \"CLIENT: LADY ADA\", and a note pinned to the wall says \"DO NOT WIND PAST MIDNIGHT\". Moody chiaroscuro, shallow depth of field, extremely fine mechanical detail.",
+    "base": {
+      "id": "P03",
+      "title": "Victorian Automaton Repair",
+      "seed": 103,
+      "prompt": "A richly detailed Victorian workshop where a brass clockwork automaton is being repaired under green banker lamps, with tiny gears, pearl inlays, oiled leather belts, smoke from a soldering iron, magnifying glass distortion, and handwritten labels everywhere. The main blueprint title must read \"AUTOMATON HAND ASSEMBLY REV. C\", a drawer label says \"SPRINGS / EYES / MEMORY CAMS\", a dangling tag says \"CLIENT: LADY ADA\", and a note pinned to the wall says \"DO NOT WIND PAST MIDNIGHT\". Moody chiaroscuro, shallow depth of field, extremely fine mechanical detail.",
+      "image": "assets/comparison/p03_original.png",
+      "time_s": 2.901,
+      "peak_allocated_gb": 23.244,
+      "peak_reserved_gb": 25.438,
+      "end_allocated_gb": 20.841,
+      "end_reserved_gb": 25.438
+    },
+    "quant": {
+      "id": "P03",
+      "title": "Victorian Automaton Repair",
+      "seed": 103,
+      "prompt": "A richly detailed Victorian workshop where a brass clockwork automaton is being repaired under green banker lamps, with tiny gears, pearl inlays, oiled leather belts, smoke from a soldering iron, magnifying glass distortion, and handwritten labels everywhere. The main blueprint title must read \"AUTOMATON HAND ASSEMBLY REV. C\", a drawer label says \"SPRINGS / EYES / MEMORY CAMS\", a dangling tag says \"CLIENT: LADY ADA\", and a note pinned to the wall says \"DO NOT WIND PAST MIDNIGHT\". Moody chiaroscuro, shallow depth of field, extremely fine mechanical detail.",
+      "image": "assets/comparison/p03_sdnq_uint_static.png",
+      "time_s": 2.846,
+      "peak_allocated_gb": 17.686,
+      "peak_reserved_gb": 19.744,
+      "end_allocated_gb": 15.283,
+      "end_reserved_gb": 19.744
+    },
+    "speed_delta_pct": -1.9
+  },
+  {
+    "id": "P04",
+    "title": "Mars Greenhouse Control Room",
+    "seed": 104,
+    "prompt": "A believable Mars greenhouse control room at dawn, red dust outside the curved windows, rows of tomatoes and dwarf wheat under violet grow lights, condensation on transparent tubes, a tired botanist reflected in a touchscreen, and several readable UI panels in English: \"OXYGEN LOOP STABLE\", \"WATER RECOVERY 98.4%\", \"SECTOR C: POLLINATION DRONES ACTIVE\", and a sticky note saying \"Tell Earth the basil survived\". Technical but warm, high resolution, realistic sci-fi, detailed glass and plant textures.",
+    "base": {
+      "id": "P04",
+      "title": "Mars Greenhouse Control Room",
+      "seed": 104,
+      "prompt": "A believable Mars greenhouse control room at dawn, red dust outside the curved windows, rows of tomatoes and dwarf wheat under violet grow lights, condensation on transparent tubes, a tired botanist reflected in a touchscreen, and several readable UI panels in English: \"OXYGEN LOOP STABLE\", \"WATER RECOVERY 98.4%\", \"SECTOR C: POLLINATION DRONES ACTIVE\", and a sticky note saying \"Tell Earth the basil survived\". Technical but warm, high resolution, realistic sci-fi, detailed glass and plant textures.",
+      "image": "assets/comparison/p04_original.png",
+      "time_s": 1.186,
+      "peak_allocated_gb": 23.242,
+      "peak_reserved_gb": 25.438,
+      "end_allocated_gb": 20.841,
+      "end_reserved_gb": 25.438
+    },
+    "quant": {
+      "id": "P04",
+      "title": "Mars Greenhouse Control Room",
+      "seed": 104,
+      "prompt": "A believable Mars greenhouse control room at dawn, red dust outside the curved windows, rows of tomatoes and dwarf wheat under violet grow lights, condensation on transparent tubes, a tired botanist reflected in a touchscreen, and several readable UI panels in English: \"OXYGEN LOOP STABLE\", \"WATER RECOVERY 98.4%\", \"SECTOR C: POLLINATION DRONES ACTIVE\", and a sticky note saying \"Tell Earth the basil survived\". Technical but warm, high resolution, realistic sci-fi, detailed glass and plant textures.",
+      "image": "assets/comparison/p04_sdnq_uint_static.png",
+      "time_s": 2.846,
+      "peak_allocated_gb": 17.684,
+      "peak_reserved_gb": 19.725,
+      "end_allocated_gb": 15.283,
+      "end_reserved_gb": 19.725
+    },
+    "speed_delta_pct": 140.0
+  },
+  {
+    "id": "P05",
+    "title": "Lost Railway Poster Wall",
+    "seed": 105,
+    "prompt": "An abandoned underground railway platform turned into an accidental archive of travel posters, peeling ceramic tiles, puddles reflecting amber emergency lights, old suitcases, vines growing through cracked concrete, and five large posters with distinct readable titles: \"THE NORTHERN COMET EXPRESS\", \"SLEEPER TO ISTANBUL\", \"MIDNIGHT PLATFORM 7\", \"COASTAL ROUTE REOPENING SOON\", and \"KEEP YOUR TICKET VISIBLE\". Cinematic composition, wet surfaces, layered typography, realistic grime, strong perspective down the tracks.",
+    "base": {
+      "id": "P05",
+      "title": "Lost Railway Poster Wall",
+      "seed": 105,
+      "prompt": "An abandoned underground railway platform turned into an accidental archive of travel posters, peeling ceramic tiles, puddles reflecting amber emergency lights, old suitcases, vines growing through cracked concrete, and five large posters with distinct readable titles: \"THE NORTHERN COMET EXPRESS\", \"SLEEPER TO ISTANBUL\", \"MIDNIGHT PLATFORM 7\", \"COASTAL ROUTE REOPENING SOON\", and \"KEEP YOUR TICKET VISIBLE\". Cinematic composition, wet surfaces, layered typography, realistic grime, strong perspective down the tracks.",
+      "image": "assets/comparison/p05_original.png",
+      "time_s": 1.185,
+      "peak_allocated_gb": 23.242,
+      "peak_reserved_gb": 25.438,
+      "end_allocated_gb": 20.841,
+      "end_reserved_gb": 25.438
+    },
+    "quant": {
+      "id": "P05",
+      "title": "Lost Railway Poster Wall",
+      "seed": 105,
+      "prompt": "An abandoned underground railway platform turned into an accidental archive of travel posters, peeling ceramic tiles, puddles reflecting amber emergency lights, old suitcases, vines growing through cracked concrete, and five large posters with distinct readable titles: \"THE NORTHERN COMET EXPRESS\", \"SLEEPER TO ISTANBUL\", \"MIDNIGHT PLATFORM 7\", \"COASTAL ROUTE REOPENING SOON\", and \"KEEP YOUR TICKET VISIBLE\". Cinematic composition, wet surfaces, layered typography, realistic grime, strong perspective down the tracks.",
+      "image": "assets/comparison/p05_sdnq_uint_static.png",
+      "time_s": 2.877,
+      "peak_allocated_gb": 17.684,
+      "peak_reserved_gb": 19.725,
+      "end_allocated_gb": 15.283,
+      "end_reserved_gb": 19.725
+    },
+    "speed_delta_pct": 142.8
+  },
+  {
+    "id": "P06",
+    "title": "Miniature Courtroom Diorama",
+    "seed": 106,
+    "prompt": "A hyperreal macro photograph of a miniature courtroom diorama built inside an antique wooden drawer, with tiny judge bench, brass lamps, dust motes, paper exhibits smaller than postage stamps, a mouse-sized witness chair, and readable text on tiny documents: a case file labeled \"CASE 1842-B: THE MISSING ORRERY\", an evidence tag saying \"EXHIBIT C\", a court calendar reading \"HEARING AT 9:30\", and a placard on the judge bench saying \"TRUTH IN SMALL THINGS\". Macro lens, tactile materials, careful scale cues.",
+    "base": {
+      "id": "P06",
+      "title": "Miniature Courtroom Diorama",
+      "seed": 106,
+      "prompt": "A hyperreal macro photograph of a miniature courtroom diorama built inside an antique wooden drawer, with tiny judge bench, brass lamps, dust motes, paper exhibits smaller than postage stamps, a mouse-sized witness chair, and readable text on tiny documents: a case file labeled \"CASE 1842-B: THE MISSING ORRERY\", an evidence tag saying \"EXHIBIT C\", a court calendar reading \"HEARING AT 9:30\", and a placard on the judge bench saying \"TRUTH IN SMALL THINGS\". Macro lens, tactile materials, careful scale cues.",
+      "image": "assets/comparison/p06_original.png",
+      "time_s": 1.186,
+      "peak_allocated_gb": 23.244,
+      "peak_reserved_gb": 25.438,
+      "end_allocated_gb": 20.841,
+      "end_reserved_gb": 25.438
+    },
+    "quant": {
+      "id": "P06",
+      "title": "Miniature Courtroom Diorama",
+      "seed": 106,
+      "prompt": "A hyperreal macro photograph of a miniature courtroom diorama built inside an antique wooden drawer, with tiny judge bench, brass lamps, dust motes, paper exhibits smaller than postage stamps, a mouse-sized witness chair, and readable text on tiny documents: a case file labeled \"CASE 1842-B: THE MISSING ORRERY\", an evidence tag saying \"EXHIBIT C\", a court calendar reading \"HEARING AT 9:30\", and a placard on the judge bench saying \"TRUTH IN SMALL THINGS\". Macro lens, tactile materials, careful scale cues.",
+      "image": "assets/comparison/p06_sdnq_uint_static.png",
+      "time_s": 2.836,
+      "peak_allocated_gb": 17.686,
+      "peak_reserved_gb": 19.744,
+      "end_allocated_gb": 15.283,
+      "end_reserved_gb": 19.744
+    },
+    "speed_delta_pct": 139.1
+  },
+  {
+    "id": "P07",
+    "title": "Rainy Seoul Book Cafe",
+    "seed": 107,
+    "prompt": "A cozy but complex rainy evening scene in a narrow Seoul book cafe, viewed through a window covered in raindrops, shelves packed with art books, two students annotating a map, a barista steaming milk, warm tungsten light, street reflections, and multiple readable English text elements: a chalkboard says \"TONIGHT: QUIET READING CLUB\", a receipt says \"OAT LATTE / CINNAMON BUN\", a book spine says \"ARCHITECTURE OF DREAMS\", and a window sticker says \"OPEN UNTIL THE LAST TRAIN\". Photorealistic, cinematic, intricate reflections.",
+    "base": {
+      "id": "P07",
+      "title": "Rainy Seoul Book Cafe",
+      "seed": 107,
+      "prompt": "A cozy but complex rainy evening scene in a narrow Seoul book cafe, viewed through a window covered in raindrops, shelves packed with art books, two students annotating a map, a barista steaming milk, warm tungsten light, street reflections, and multiple readable English text elements: a chalkboard says \"TONIGHT: QUIET READING CLUB\", a receipt says \"OAT LATTE / CINNAMON BUN\", a book spine says \"ARCHITECTURE OF DREAMS\", and a window sticker says \"OPEN UNTIL THE LAST TRAIN\". Photorealistic, cinematic, intricate reflections.",
+      "image": "assets/comparison/p07_original.png",
+      "time_s": 1.243,
+      "peak_allocated_gb": 23.244,
+      "peak_reserved_gb": 25.438,
+      "end_allocated_gb": 20.841,
+      "end_reserved_gb": 25.438
+    },
+    "quant": {
+      "id": "P07",
+      "title": "Rainy Seoul Book Cafe",
+      "seed": 107,
+      "prompt": "A cozy but complex rainy evening scene in a narrow Seoul book cafe, viewed through a window covered in raindrops, shelves packed with art books, two students annotating a map, a barista steaming milk, warm tungsten light, street reflections, and multiple readable English text elements: a chalkboard says \"TONIGHT: QUIET READING CLUB\", a receipt says \"OAT LATTE / CINNAMON BUN\", a book spine says \"ARCHITECTURE OF DREAMS\", and a window sticker says \"OPEN UNTIL THE LAST TRAIN\". Photorealistic, cinematic, intricate reflections.",
+      "image": "assets/comparison/p07_sdnq_uint_static.png",
+      "time_s": 2.835,
+      "peak_allocated_gb": 17.686,
+      "peak_reserved_gb": 19.744,
+      "end_allocated_gb": 15.283,
+      "end_reserved_gb": 19.744
+    },
+    "speed_delta_pct": 128.1
+  },
+  {
+    "id": "P08",
+    "title": "Oceanographic Expedition Map",
+    "seed": 108,
+    "prompt": "A dramatic captain's table aboard a storm-tossed oceanographic research vessel, with a wet nautical chart, brass dividers, sonar printouts, bioluminescent plankton glowing in a glass jar, a cracked tablet, and readable labels distributed across the image: \"TRENCH SURVEY LINE B\", \"DEPTH 10,928m\", \"ROV SIGNAL WEAK\", \"SAMPLE: BLUE VENT WATER\", and a torn note saying \"If the lights pulse twice, turn back\". High detail, realistic water droplets, dark blue-green atmosphere, sharp text.",
+    "base": {
+      "id": "P08",
+      "title": "Oceanographic Expedition Map",
+      "seed": 108,
+      "prompt": "A dramatic captain's table aboard a storm-tossed oceanographic research vessel, with a wet nautical chart, brass dividers, sonar printouts, bioluminescent plankton glowing in a glass jar, a cracked tablet, and readable labels distributed across the image: \"TRENCH SURVEY LINE B\", \"DEPTH 10,928m\", \"ROV SIGNAL WEAK\", \"SAMPLE: BLUE VENT WATER\", and a torn note saying \"If the lights pulse twice, turn back\". High detail, realistic water droplets, dark blue-green atmosphere, sharp text.",
+      "image": "assets/comparison/p08_original.png",
+      "time_s": 1.181,
+      "peak_allocated_gb": 23.244,
+      "peak_reserved_gb": 25.438,
+      "end_allocated_gb": 20.841,
+      "end_reserved_gb": 25.438
+    },
+    "quant": {
+      "id": "P08",
+      "title": "Oceanographic Expedition Map",
+      "seed": 108,
+      "prompt": "A dramatic captain's table aboard a storm-tossed oceanographic research vessel, with a wet nautical chart, brass dividers, sonar printouts, bioluminescent plankton glowing in a glass jar, a cracked tablet, and readable labels distributed across the image: \"TRENCH SURVEY LINE B\", \"DEPTH 10,928m\", \"ROV SIGNAL WEAK\", \"SAMPLE: BLUE VENT WATER\", and a torn note saying \"If the lights pulse twice, turn back\". High detail, realistic water droplets, dark blue-green atmosphere, sharp text.",
+      "image": "assets/comparison/p08_sdnq_uint_static.png",
+      "time_s": 2.847,
+      "peak_allocated_gb": 17.686,
+      "peak_reserved_gb": 19.744,
+      "end_allocated_gb": 15.283,
+      "end_reserved_gb": 19.744
+    },
+    "speed_delta_pct": 141.1
+  },
+  {
+    "id": "P09",
+    "title": "Renaissance Lab Notebook",
+    "seed": 109,
+    "prompt": "An alternate-history Renaissance laboratory where an astronomer-painter is combining oil pigments with early electrical apparatus, with celestial globes, copper coils, stained glass sunlight, anatomical sketches, a half-finished portrait, and Latin-English notebook text visible on several pages: \"LIGHT STUDY: BLUE VERDITER\", \"GALVANIC TEST NO. 8\", \"VENUS RISES BEFORE DAWN\", and a folded letter sealed in wax reading \"FOR THE WORKSHOP MASTER ONLY\". Painterly realism, ornate detail, coherent objects, readable calligraphy.",
+    "base": {
+      "id": "P09",
+      "title": "Renaissance Lab Notebook",
+      "seed": 109,
+      "prompt": "An alternate-history Renaissance laboratory where an astronomer-painter is combining oil pigments with early electrical apparatus, with celestial globes, copper coils, stained glass sunlight, anatomical sketches, a half-finished portrait, and Latin-English notebook text visible on several pages: \"LIGHT STUDY: BLUE VERDITER\", \"GALVANIC TEST NO. 8\", \"VENUS RISES BEFORE DAWN\", and a folded letter sealed in wax reading \"FOR THE WORKSHOP MASTER ONLY\". Painterly realism, ornate detail, coherent objects, readable calligraphy.",
+      "image": "assets/comparison/p09_original.png",
+      "time_s": 1.18,
+      "peak_allocated_gb": 23.242,
+      "peak_reserved_gb": 25.438,
+      "end_allocated_gb": 20.841,
+      "end_reserved_gb": 25.438
+    },
+    "quant": {
+      "id": "P09",
+      "title": "Renaissance Lab Notebook",
+      "seed": 109,
+      "prompt": "An alternate-history Renaissance laboratory where an astronomer-painter is combining oil pigments with early electrical apparatus, with celestial globes, copper coils, stained glass sunlight, anatomical sketches, a half-finished portrait, and Latin-English notebook text visible on several pages: \"LIGHT STUDY: BLUE VERDITER\", \"GALVANIC TEST NO. 8\", \"VENUS RISES BEFORE DAWN\", and a folded letter sealed in wax reading \"FOR THE WORKSHOP MASTER ONLY\". Painterly realism, ornate detail, coherent objects, readable calligraphy.",
+      "image": "assets/comparison/p09_sdnq_uint_static.png",
+      "time_s": 2.855,
+      "peak_allocated_gb": 17.684,
+      "peak_reserved_gb": 19.719,
+      "end_allocated_gb": 15.283,
+      "end_reserved_gb": 19.719
+    },
+    "speed_delta_pct": 141.9
+  },
+  {
+    "id": "P10",
+    "title": "Russian Provincial Print Shop",
+    "seed": 110,
+    "prompt": "Сложная фотореалистичная сцена в старой провинциальной типографии поздним зимним вечером: за большим деревянным столом лежат металлические литеры, корректурные листы, линейки, чашка крепкого чая, заснеженное окно, тусклая лампа и следы типографской краски на пальцах наборщика. На разных элементах изображения должен быть длинный и хорошо читаемый русский текст: на вывеске над дверью написано \"ТИПОГРАФИЯ СЕВЕРНЫЙ ЛИСТОК\", на корректуре заголовок \"СРОЧНО В НОМЕР: ГОРОДСКОЙ СОВЕТ ОТКРЫВАЕТ НОВУЮ БИБЛИОТЕКУ\", на маленькой записке фраза \"Проверить букву Ё во втором абзаце\", а на календаре дата \"Пятница, 24 января\". Много бытовых деталей, глубокие тени, реалистичная кириллица, никакой размытой каши вместо текста.",
+    "base": {
+      "id": "P10",
+      "title": "Russian Provincial Print Shop",
+      "seed": 110,
+      "prompt": "Сложная фотореалистичная сцена в старой провинциальной типографии поздним зимним вечером: за большим деревянным столом лежат металлические литеры, корректурные листы, линейки, чашка крепкого чая, заснеженное окно, тусклая лампа и следы типографской краски на пальцах наборщика. На разных элементах изображения должен быть длинный и хорошо читаемый русский текст: на вывеске над дверью написано \"ТИПОГРАФИЯ СЕВЕРНЫЙ ЛИСТОК\", на корректуре заголовок \"СРОЧНО В НОМЕР: ГОРОДСКОЙ СОВЕТ ОТКРЫВАЕТ НОВУЮ БИБЛИОТЕКУ\", на маленькой записке фраза \"Проверить букву Ё во втором абзаце\", а на календаре дата \"Пятница, 24 января\". Много бытовых деталей, глубокие тени, реалистичная кириллица, никакой размытой каши вместо текста.",
+      "image": "assets/comparison/p10_original.png",
+      "time_s": 3.749,
+      "peak_allocated_gb": 23.252,
+      "peak_reserved_gb": 25.459,
+      "end_allocated_gb": 20.841,
+      "end_reserved_gb": 25.459
+    },
+    "quant": {
+      "id": "P10",
+      "title": "Russian Provincial Print Shop",
+      "seed": 110,
+      "prompt": "Сложная фотореалистичная сцена в старой провинциальной типографии поздним зимним вечером: за большим деревянным столом лежат металлические литеры, корректурные листы, линейки, чашка крепкого чая, заснеженное окно, тусклая лампа и следы типографской краски на пальцах наборщика. На разных элементах изображения должен быть длинный и хорошо читаемый русский текст: на вывеске над дверью написано \"ТИПОГРАФИЯ СЕВЕРНЫЙ ЛИСТОК\", на корректуре заголовок \"СРОЧНО В НОМЕР: ГОРОДСКОЙ СОВЕТ ОТКРЫВАЕТ НОВУЮ БИБЛИОТЕКУ\", на маленькой записке фраза \"Проверить букву Ё во втором абзаце\", а на календаре дата \"Пятница, 24 января\". Много бытовых деталей, глубокие тени, реалистичная кириллица, никакой размытой каши вместо текста.",
+      "image": "assets/comparison/p10_sdnq_uint_static.png",
+      "time_s": 2.888,
+      "peak_allocated_gb": 17.695,
+      "peak_reserved_gb": 19.766,
+      "end_allocated_gb": 15.283,
+      "end_reserved_gb": 19.766
+    },
+    "speed_delta_pct": -23.0
+  }
+]

model_index.json

@@ -0,0 +1,25 @@
+{
+  "_class_name": "LensPipeline",
+  "_diffusers_version": "0.38.0",
+  "_name_or_path": "microsoft/Lens-Turbo",
+  "scheduler": [
+    "diffusers",
+    "FlowMatchEulerDiscreteScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "LensGptOssEncoder"
+  ],
+  "tokenizer": [
+    "transformers",
+    "PreTrainedTokenizerFast"
+  ],
+  "transformer": [
+    "diffusers",
+    "LensTransformer2DModel"
+  ],
+  "vae": [
+    "diffusers",
+    "AutoencoderKLFlux2"
+  ]
+}

scheduler/scheduler_config.json

@@ -0,0 +1,18 @@
+{
+  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_diffusers_version": "0.38.0",
+  "base_image_seq_len": 256,
+  "base_shift": 0.5,
+  "invert_sigmas": false,
+  "max_image_seq_len": 4096,
+  "max_shift": 1.15,
+  "num_train_timesteps": 1000,
+  "shift": 3.0,
+  "shift_terminal": null,
+  "stochastic_sampling": false,
+  "time_shift_type": "exponential",
+  "use_beta_sigmas": false,
+  "use_dynamic_shifting": true,
+  "use_exponential_sigmas": false,
+  "use_karras_sigmas": false
+}

sdnq_quantization_summary.json

@@ -0,0 +1,22 @@
+{
+  "source_model": "microsoft/Lens-Turbo",
+  "method": "SDNQ Uint Static",
+  "scope": "transformer only",
+  "text_encoder": "kept as provided by source checkpoint (GPT-OSS / MXFP4 path)",
+  "vae": "kept in bfloat16",
+  "config": {
+    "weights_dtype": "uint4",
+    "quantized_matmul_dtype": "int8",
+    "group_size": "0",
+    "use_static_quantization": "True",
+    "use_dynamic_quantization": "False",
+    "use_quantized_matmul": "True",
+    "use_svd": "False",
+    "use_hadamard": "False",
+    "quant_conv": "False",
+    "quant_embedding": "False",
+    "dequantize_fp32": "False",
+    "quantization_device": "cuda",
+    "return_device": "cuda"
+  }
+}

text_encoder/config.json

@@ -0,0 +1,78 @@
+{
+  "architectures": [
+    "LensGptOssEncoder"
+  ],
+  "attention_bias": true,
+  "attention_dropout": 0.0,
+  "bos_token_id": null,
+  "dtype": "bfloat16",
+  "eos_token_id": 200002,
+  "experts_per_token": 4,
+  "head_dim": 64,
+  "hidden_act": "silu",
+  "hidden_size": 2880,
+  "initial_context_length": 4096,
+  "initializer_range": 0.02,
+  "intermediate_size": 2880,
+  "layer_types": [
+    "sliding_attention",
+    "full_attention",
+    "sliding_attention",
+    "full_attention",
+    "sliding_attention",
+    "full_attention",
+    "sliding_attention",
+    "full_attention",
+    "sliding_attention",
+    "full_attention",
+    "sliding_attention",
+    "full_attention",
+    "sliding_attention",
+    "full_attention",
+    "sliding_attention",
+    "full_attention",
+    "sliding_attention",
+    "full_attention",
+    "sliding_attention",
+    "full_attention",
+    "sliding_attention",
+    "full_attention",
+    "sliding_attention",
+    "full_attention"
+  ],
+  "max_position_embeddings": 131072,
+  "model_type": "gpt_oss",
+  "num_attention_heads": 64,
+  "num_experts_per_tok": 4,
+  "num_hidden_layers": 24,
+  "num_key_value_heads": 8,
+  "num_local_experts": 32,
+  "output_router_logits": false,
+  "pad_token_id": 199999,
+  "quantization_config": {
+    "modules_to_not_convert": [
+      "model.layers.*.self_attn",
+      "model.layers.*.mlp.router",
+      "model.embed_tokens",
+      "lm_head"
+    ],
+    "quant_method": "mxfp4"
+  },
+  "rms_norm_eps": 1e-05,
+  "rope_parameters": {
+    "beta_fast": 32.0,
+    "beta_slow": 1.0,
+    "factor": 32.0,
+    "original_max_position_embeddings": 4096,
+    "rope_theta": 150000,
+    "rope_type": "yarn",
+    "truncate": false
+  },
+  "router_aux_loss_coef": 0.9,
+  "sliding_window": 128,
+  "swiglu_limit": 7.0,
+  "tie_word_embeddings": false,
+  "transformers_version": "5.8.0",
+  "use_cache": true,
+  "vocab_size": 201088
+}

text_encoder/generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "eos_token_id": 200002,
+  "pad_token_id": 199999,
+  "transformers_version": "5.8.0",
+  "use_cache": true
+}

text_encoder/model-00001-of-00004.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:417c7e4acfe883a1b65991f8be06f1bcbc09cf8b4d86c92330a0ad6b58b75029
+size 3890907096

text_encoder/model-00002-of-00004.safetensors

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

text_encoder/model-00003-of-00004.safetensors

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

text_encoder/model-00004-of-00004.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:48bbcc91f726dc52c0577b773a00222ed89c17a8c947b20570c514fa4b93f1a5
+size 2244982120

text_encoder/model.safetensors.index.json

@@ -0,0 +1,467 @@
+{
+  "metadata": {
+    "total_parameters": 1804459584,
+    "total_size": 13774535808
+  },
+  "weight_map": {
+    "lm_head.weight": "model-00001-of-00004.safetensors",
+    "model.embed_tokens.weight": "model-00001-of-00004.safetensors",
+    "model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
+    "model.layers.0.mlp.experts.down_proj_bias": "model-00001-of-00004.safetensors",
+    "model.layers.0.mlp.experts.down_proj_blocks": "model-00001-of-00004.safetensors",
+    "model.layers.0.mlp.experts.down_proj_scales": "model-00001-of-00004.safetensors",
+    "model.layers.0.mlp.experts.gate_up_proj_bias": "model-00001-of-00004.safetensors",
+    "model.layers.0.mlp.experts.gate_up_proj_blocks": "model-00001-of-00004.safetensors",
+    "model.layers.0.mlp.experts.gate_up_proj_scales": "model-00001-of-00004.safetensors",
+    "model.layers.0.mlp.router.bias": "model-00001-of-00004.safetensors",
+    "model.layers.0.mlp.router.weight": "model-00001-of-00004.safetensors",
+    "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
+    "model.layers.0.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.0.self_attn.o_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.0.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.0.self_attn.sinks": "model-00001-of-00004.safetensors",
+    "model.layers.0.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors",
+    "model.layers.1.mlp.experts.down_proj_bias": "model-00001-of-00004.safetensors",
+    "model.layers.1.mlp.experts.down_proj_blocks": "model-00001-of-00004.safetensors",
+    "model.layers.1.mlp.experts.down_proj_scales": "model-00001-of-00004.safetensors",
+    "model.layers.1.mlp.experts.gate_up_proj_bias": "model-00001-of-00004.safetensors",
+    "model.layers.1.mlp.experts.gate_up_proj_blocks": "model-00001-of-00004.safetensors",
+    "model.layers.1.mlp.experts.gate_up_proj_scales": "model-00001-of-00004.safetensors",
+    "model.layers.1.mlp.router.bias": "model-00001-of-00004.safetensors",
+    "model.layers.1.mlp.router.weight": "model-00001-of-00004.safetensors",
+    "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
+    "model.layers.1.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.1.self_attn.o_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.1.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.1.self_attn.sinks": "model-00001-of-00004.safetensors",
+    "model.layers.1.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.10.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.10.mlp.experts.down_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.10.mlp.experts.down_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.10.mlp.experts.down_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.10.mlp.experts.gate_up_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.10.mlp.experts.gate_up_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.10.mlp.experts.gate_up_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.10.mlp.router.bias": "model-00002-of-00004.safetensors",
+    "model.layers.10.mlp.router.weight": "model-00002-of-00004.safetensors",
+    "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.10.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.10.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.10.self_attn.o_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.10.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.10.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.10.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.10.self_attn.sinks": "model-00002-of-00004.safetensors",
+    "model.layers.10.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.10.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.11.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.11.mlp.experts.down_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.11.mlp.experts.down_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.11.mlp.experts.down_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.11.mlp.experts.gate_up_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.11.mlp.experts.gate_up_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.11.mlp.experts.gate_up_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.11.mlp.router.bias": "model-00003-of-00004.safetensors",
+    "model.layers.11.mlp.router.weight": "model-00003-of-00004.safetensors",
+    "model.layers.11.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.11.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.11.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.11.self_attn.o_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.11.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.11.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.11.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.11.self_attn.sinks": "model-00003-of-00004.safetensors",
+    "model.layers.11.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.11.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.12.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.12.mlp.experts.down_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.12.mlp.experts.down_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.12.mlp.experts.down_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.12.mlp.experts.gate_up_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.12.mlp.experts.gate_up_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.12.mlp.experts.gate_up_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.12.mlp.router.bias": "model-00003-of-00004.safetensors",
+    "model.layers.12.mlp.router.weight": "model-00003-of-00004.safetensors",
+    "model.layers.12.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.12.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.12.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.12.self_attn.o_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.12.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.12.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.12.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.12.self_attn.sinks": "model-00003-of-00004.safetensors",
+    "model.layers.12.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.12.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.13.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.13.mlp.experts.down_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.13.mlp.experts.down_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.13.mlp.experts.down_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.13.mlp.experts.gate_up_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.13.mlp.experts.gate_up_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.13.mlp.experts.gate_up_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.13.mlp.router.bias": "model-00003-of-00004.safetensors",
+    "model.layers.13.mlp.router.weight": "model-00003-of-00004.safetensors",
+    "model.layers.13.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.13.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.13.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.13.self_attn.o_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.13.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.13.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.13.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.13.self_attn.sinks": "model-00003-of-00004.safetensors",
+    "model.layers.13.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.13.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.14.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.14.mlp.experts.down_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.14.mlp.experts.down_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.14.mlp.experts.down_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.14.mlp.experts.gate_up_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.14.mlp.experts.gate_up_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.14.mlp.experts.gate_up_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.14.mlp.router.bias": "model-00003-of-00004.safetensors",
+    "model.layers.14.mlp.router.weight": "model-00003-of-00004.safetensors",
+    "model.layers.14.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.14.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.14.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.14.self_attn.o_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.14.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.14.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.14.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.14.self_attn.sinks": "model-00003-of-00004.safetensors",
+    "model.layers.14.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.14.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.15.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.15.mlp.experts.down_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.15.mlp.experts.down_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.15.mlp.experts.down_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.15.mlp.experts.gate_up_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.15.mlp.experts.gate_up_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.15.mlp.experts.gate_up_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.15.mlp.router.bias": "model-00003-of-00004.safetensors",
+    "model.layers.15.mlp.router.weight": "model-00003-of-00004.safetensors",
+    "model.layers.15.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.15.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.15.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.15.self_attn.o_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.15.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.15.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.15.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.15.self_attn.sinks": "model-00003-of-00004.safetensors",
+    "model.layers.15.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.15.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.16.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.16.mlp.experts.down_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.16.mlp.experts.down_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.16.mlp.experts.down_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.16.mlp.experts.gate_up_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.16.mlp.experts.gate_up_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.16.mlp.experts.gate_up_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.16.mlp.router.bias": "model-00003-of-00004.safetensors",
+    "model.layers.16.mlp.router.weight": "model-00003-of-00004.safetensors",
+    "model.layers.16.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.16.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.16.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.16.self_attn.o_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.16.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.16.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.16.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.16.self_attn.sinks": "model-00003-of-00004.safetensors",
+    "model.layers.16.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.16.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.17.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.17.mlp.experts.down_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.17.mlp.experts.down_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.17.mlp.experts.down_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.17.mlp.experts.gate_up_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.17.mlp.experts.gate_up_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.17.mlp.experts.gate_up_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.17.mlp.router.bias": "model-00003-of-00004.safetensors",
+    "model.layers.17.mlp.router.weight": "model-00003-of-00004.safetensors",
+    "model.layers.17.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.17.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.17.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.17.self_attn.o_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.17.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.17.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.17.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.17.self_attn.sinks": "model-00003-of-00004.safetensors",
+    "model.layers.17.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.17.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.18.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.18.mlp.experts.down_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.18.mlp.experts.down_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.18.mlp.experts.down_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.18.mlp.experts.gate_up_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.18.mlp.experts.gate_up_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.18.mlp.experts.gate_up_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.18.mlp.router.bias": "model-00003-of-00004.safetensors",
+    "model.layers.18.mlp.router.weight": "model-00003-of-00004.safetensors",
+    "model.layers.18.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.18.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.18.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.18.self_attn.o_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.18.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.18.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.18.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.18.self_attn.sinks": "model-00003-of-00004.safetensors",
+    "model.layers.18.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
+    "model.layers.18.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
+    "model.layers.19.input_layernorm.weight": "model-00003-of-00004.safetensors",
+    "model.layers.19.mlp.experts.down_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.19.mlp.experts.down_proj_blocks": "model-00003-of-00004.safetensors",
+    "model.layers.19.mlp.experts.down_proj_scales": "model-00003-of-00004.safetensors",
+    "model.layers.19.mlp.experts.gate_up_proj_bias": "model-00003-of-00004.safetensors",
+    "model.layers.19.mlp.experts.gate_up_proj_blocks": "model-00004-of-00004.safetensors",
+    "model.layers.19.mlp.experts.gate_up_proj_scales": "model-00004-of-00004.safetensors",
+    "model.layers.19.mlp.router.bias": "model-00004-of-00004.safetensors",
+    "model.layers.19.mlp.router.weight": "model-00004-of-00004.safetensors",
+    "model.layers.19.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
+    "model.layers.19.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.19.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.19.self_attn.o_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.19.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.19.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.19.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.19.self_attn.sinks": "model-00004-of-00004.safetensors",
+    "model.layers.19.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.19.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.2.input_layernorm.weight": "model-00001-of-00004.safetensors",
+    "model.layers.2.mlp.experts.down_proj_bias": "model-00001-of-00004.safetensors",
+    "model.layers.2.mlp.experts.down_proj_blocks": "model-00001-of-00004.safetensors",
+    "model.layers.2.mlp.experts.down_proj_scales": "model-00001-of-00004.safetensors",
+    "model.layers.2.mlp.experts.gate_up_proj_bias": "model-00001-of-00004.safetensors",
+    "model.layers.2.mlp.experts.gate_up_proj_blocks": "model-00001-of-00004.safetensors",
+    "model.layers.2.mlp.experts.gate_up_proj_scales": "model-00001-of-00004.safetensors",
+    "model.layers.2.mlp.router.bias": "model-00001-of-00004.safetensors",
+    "model.layers.2.mlp.router.weight": "model-00001-of-00004.safetensors",
+    "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
+    "model.layers.2.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.2.self_attn.o_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.2.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.2.self_attn.sinks": "model-00001-of-00004.safetensors",
+    "model.layers.2.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
+    "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
+    "model.layers.20.input_layernorm.weight": "model-00004-of-00004.safetensors",
+    "model.layers.20.mlp.experts.down_proj_bias": "model-00004-of-00004.safetensors",
+    "model.layers.20.mlp.experts.down_proj_blocks": "model-00004-of-00004.safetensors",
+    "model.layers.20.mlp.experts.down_proj_scales": "model-00004-of-00004.safetensors",
+    "model.layers.20.mlp.experts.gate_up_proj_bias": "model-00004-of-00004.safetensors",
+    "model.layers.20.mlp.experts.gate_up_proj_blocks": "model-00004-of-00004.safetensors",
+    "model.layers.20.mlp.experts.gate_up_proj_scales": "model-00004-of-00004.safetensors",
+    "model.layers.20.mlp.router.bias": "model-00004-of-00004.safetensors",
+    "model.layers.20.mlp.router.weight": "model-00004-of-00004.safetensors",
+    "model.layers.20.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
+    "model.layers.20.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.20.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.20.self_attn.o_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.20.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.20.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.20.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.20.self_attn.sinks": "model-00004-of-00004.safetensors",
+    "model.layers.20.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.20.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.21.input_layernorm.weight": "model-00004-of-00004.safetensors",
+    "model.layers.21.mlp.experts.down_proj_bias": "model-00004-of-00004.safetensors",
+    "model.layers.21.mlp.experts.down_proj_blocks": "model-00004-of-00004.safetensors",
+    "model.layers.21.mlp.experts.down_proj_scales": "model-00004-of-00004.safetensors",
+    "model.layers.21.mlp.experts.gate_up_proj_bias": "model-00004-of-00004.safetensors",
+    "model.layers.21.mlp.experts.gate_up_proj_blocks": "model-00004-of-00004.safetensors",
+    "model.layers.21.mlp.experts.gate_up_proj_scales": "model-00004-of-00004.safetensors",
+    "model.layers.21.mlp.router.bias": "model-00004-of-00004.safetensors",
+    "model.layers.21.mlp.router.weight": "model-00004-of-00004.safetensors",
+    "model.layers.21.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
+    "model.layers.21.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.21.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.21.self_attn.o_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.21.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.21.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.21.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.21.self_attn.sinks": "model-00004-of-00004.safetensors",
+    "model.layers.21.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.21.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.22.input_layernorm.weight": "model-00004-of-00004.safetensors",
+    "model.layers.22.mlp.experts.down_proj_bias": "model-00004-of-00004.safetensors",
+    "model.layers.22.mlp.experts.down_proj_blocks": "model-00004-of-00004.safetensors",
+    "model.layers.22.mlp.experts.down_proj_scales": "model-00004-of-00004.safetensors",
+    "model.layers.22.mlp.experts.gate_up_proj_bias": "model-00004-of-00004.safetensors",
+    "model.layers.22.mlp.experts.gate_up_proj_blocks": "model-00004-of-00004.safetensors",
+    "model.layers.22.mlp.experts.gate_up_proj_scales": "model-00004-of-00004.safetensors",
+    "model.layers.22.mlp.router.bias": "model-00004-of-00004.safetensors",
+    "model.layers.22.mlp.router.weight": "model-00004-of-00004.safetensors",
+    "model.layers.22.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
+    "model.layers.22.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.22.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.22.self_attn.o_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.22.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.22.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.22.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.22.self_attn.sinks": "model-00004-of-00004.safetensors",
+    "model.layers.22.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.22.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.23.input_layernorm.weight": "model-00004-of-00004.safetensors",
+    "model.layers.23.mlp.experts.down_proj_bias": "model-00004-of-00004.safetensors",
+    "model.layers.23.mlp.experts.down_proj_blocks": "model-00004-of-00004.safetensors",
+    "model.layers.23.mlp.experts.down_proj_scales": "model-00004-of-00004.safetensors",
+    "model.layers.23.mlp.experts.gate_up_proj_bias": "model-00004-of-00004.safetensors",
+    "model.layers.23.mlp.experts.gate_up_proj_blocks": "model-00004-of-00004.safetensors",
+    "model.layers.23.mlp.experts.gate_up_proj_scales": "model-00004-of-00004.safetensors",
+    "model.layers.23.mlp.router.bias": "model-00004-of-00004.safetensors",
+    "model.layers.23.mlp.router.weight": "model-00004-of-00004.safetensors",
+    "model.layers.23.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
+    "model.layers.23.self_attn.k_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.23.self_attn.k_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.23.self_attn.o_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.23.self_attn.o_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.23.self_attn.q_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.23.self_attn.q_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.23.self_attn.sinks": "model-00004-of-00004.safetensors",
+    "model.layers.23.self_attn.v_proj.bias": "model-00004-of-00004.safetensors",
+    "model.layers.23.self_attn.v_proj.weight": "model-00004-of-00004.safetensors",
+    "model.layers.3.input_layernorm.weight": "model-00001-of-00004.safetensors",
+    "model.layers.3.mlp.experts.down_proj_bias": "model-00001-of-00004.safetensors",
+    "model.layers.3.mlp.experts.down_proj_blocks": "model-00001-of-00004.safetensors",
+    "model.layers.3.mlp.experts.down_proj_scales": "model-00001-of-00004.safetensors",
+    "model.layers.3.mlp.experts.gate_up_proj_bias": "model-00001-of-00004.safetensors",
+    "model.layers.3.mlp.experts.gate_up_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.3.mlp.experts.gate_up_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.3.mlp.router.bias": "model-00002-of-00004.safetensors",
+    "model.layers.3.mlp.router.weight": "model-00002-of-00004.safetensors",
+    "model.layers.3.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.3.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.3.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.3.self_attn.o_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.3.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.3.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.3.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.3.self_attn.sinks": "model-00002-of-00004.safetensors",
+    "model.layers.3.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.3.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.4.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.4.mlp.experts.down_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.4.mlp.experts.down_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.4.mlp.experts.down_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.4.mlp.experts.gate_up_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.4.mlp.experts.gate_up_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.4.mlp.experts.gate_up_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.4.mlp.router.bias": "model-00002-of-00004.safetensors",
+    "model.layers.4.mlp.router.weight": "model-00002-of-00004.safetensors",
+    "model.layers.4.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.4.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.4.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.4.self_attn.o_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.4.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.4.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.4.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.4.self_attn.sinks": "model-00002-of-00004.safetensors",
+    "model.layers.4.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.4.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.5.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.5.mlp.experts.down_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.5.mlp.experts.down_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.5.mlp.experts.down_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.5.mlp.experts.gate_up_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.5.mlp.experts.gate_up_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.5.mlp.experts.gate_up_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.5.mlp.router.bias": "model-00002-of-00004.safetensors",
+    "model.layers.5.mlp.router.weight": "model-00002-of-00004.safetensors",
+    "model.layers.5.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.5.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.5.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.5.self_attn.o_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.5.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.5.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.5.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.5.self_attn.sinks": "model-00002-of-00004.safetensors",
+    "model.layers.5.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.5.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.6.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.6.mlp.experts.down_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.6.mlp.experts.down_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.6.mlp.experts.down_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.6.mlp.experts.gate_up_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.6.mlp.experts.gate_up_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.6.mlp.experts.gate_up_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.6.mlp.router.bias": "model-00002-of-00004.safetensors",
+    "model.layers.6.mlp.router.weight": "model-00002-of-00004.safetensors",
+    "model.layers.6.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.6.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.6.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.6.self_attn.o_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.6.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.6.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.6.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.6.self_attn.sinks": "model-00002-of-00004.safetensors",
+    "model.layers.6.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.6.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.7.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.7.mlp.experts.down_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.7.mlp.experts.down_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.7.mlp.experts.down_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.7.mlp.experts.gate_up_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.7.mlp.experts.gate_up_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.7.mlp.experts.gate_up_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.7.mlp.router.bias": "model-00002-of-00004.safetensors",
+    "model.layers.7.mlp.router.weight": "model-00002-of-00004.safetensors",
+    "model.layers.7.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.7.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.7.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.7.self_attn.o_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.7.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.7.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.7.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.7.self_attn.sinks": "model-00002-of-00004.safetensors",
+    "model.layers.7.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.7.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.8.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.8.mlp.experts.down_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.8.mlp.experts.down_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.8.mlp.experts.down_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.8.mlp.experts.gate_up_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.8.mlp.experts.gate_up_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.8.mlp.experts.gate_up_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.8.mlp.router.bias": "model-00002-of-00004.safetensors",
+    "model.layers.8.mlp.router.weight": "model-00002-of-00004.safetensors",
+    "model.layers.8.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.8.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.8.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.8.self_attn.o_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.8.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.8.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.8.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.8.self_attn.sinks": "model-00002-of-00004.safetensors",
+    "model.layers.8.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.8.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.9.input_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.9.mlp.experts.down_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.9.mlp.experts.down_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.9.mlp.experts.down_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.9.mlp.experts.gate_up_proj_bias": "model-00002-of-00004.safetensors",
+    "model.layers.9.mlp.experts.gate_up_proj_blocks": "model-00002-of-00004.safetensors",
+    "model.layers.9.mlp.experts.gate_up_proj_scales": "model-00002-of-00004.safetensors",
+    "model.layers.9.mlp.router.bias": "model-00002-of-00004.safetensors",
+    "model.layers.9.mlp.router.weight": "model-00002-of-00004.safetensors",
+    "model.layers.9.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
+    "model.layers.9.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.9.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.9.self_attn.o_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.9.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.9.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.9.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
+    "model.layers.9.self_attn.sinks": "model-00002-of-00004.safetensors",
+    "model.layers.9.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
+    "model.layers.9.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
+    "model.norm.weight": "model-00004-of-00004.safetensors"
+  }
+}

tokenizer/chat_template.jinja

@@ -0,0 +1,331 @@
+{#-
+  In addition to the normal inputs of `messages` and `tools`, this template also accepts the
+  following kwargs:
+  - "builtin_tools": A list, can contain "browser" and/or "python".
+  - "model_identity": A string that optionally describes the model identity.
+  - "reasoning_effort": A string that describes the reasoning effort, defaults to "medium".
+ #}
+
+{#- Tool Definition Rendering ============================================== #}
+{%- macro render_typescript_type(param_spec, required_params, is_nullable=false) -%}
+    {%- if param_spec.type == "array" -%}
+        {%- if param_spec['items'] -%}
+            {%- if param_spec['items']['type'] == "string" -%}
+                {{- "string[]" }}
+            {%- elif param_spec['items']['type'] == "number" -%}
+                {{- "number[]" }}
+            {%- elif param_spec['items']['type'] == "integer" -%}
+                {{- "number[]" }}
+            {%- elif param_spec['items']['type'] == "boolean" -%}
+                {{- "boolean[]" }}
+            {%- else -%}
+                {%- set inner_type = render_typescript_type(param_spec['items'], required_params) -%}
+                {%- if inner_type == "object | object" or inner_type|length > 50 -%}
+                    {{- "any[]" }}
+                {%- else -%}
+                    {{- inner_type + "[]" }}
+                {%- endif -%}
+            {%- endif -%}
+            {%- if param_spec.nullable -%}
+                {{- " | null" }}
+            {%- endif -%}
+        {%- else -%}
+            {{- "any[]" }}
+            {%- if param_spec.nullable -%}
+                {{- " | null" }}
+            {%- endif -%}
+        {%- endif -%}
+    {%- elif param_spec.type is defined and param_spec.type is iterable and param_spec.type is not string and param_spec.type is not mapping and param_spec.type[0] is defined -%}
+        {#- Handle array of types like ["object", "object"] from Union[dict, list] #}
+        {%- if param_spec.type | length > 1 -%}
+            {{- param_spec.type | join(" | ") }}
+        {%- else -%}
+            {{- param_spec.type[0] }}
+        {%- endif -%}
+    {%- elif param_spec.oneOf -%}
+        {#- Handle oneOf schemas - check for complex unions and fallback to any #}
+        {%- set has_object_variants = false -%}
+        {%- for variant in param_spec.oneOf -%}
+            {%- if variant.type == "object" -%}
+                {%- set has_object_variants = true -%}
+            {%- endif -%}
+        {%- endfor -%}
+        {%- if has_object_variants and param_spec.oneOf|length > 1 -%}
+            {{- "any" }}
+        {%- else -%}
+            {%- for variant in param_spec.oneOf -%}
+                {{- render_typescript_type(variant, required_params) -}}
+                {%- if variant.description %}
+                    {{- "// " + variant.description }}
+                {%- endif -%}
+                {%- if variant.default is defined %}
+                    {{ "// default: " + variant.default|tojson }}
+                {%- endif -%}
+                {%- if not loop.last %}
+                    {{- " | " }}
+                {% endif -%}
+            {%- endfor -%}
+        {%- endif -%}
+    {%- elif param_spec.type == "string" -%}
+        {%- if param_spec.enum -%}
+            {{- '"' + param_spec.enum|join('" | "') + '"' -}}
+        {%- else -%}
+            {{- "string" }}
+            {%- if param_spec.nullable %}
+                {{- " | null" }}
+            {%- endif -%}
+        {%- endif -%}
+    {%- elif param_spec.type == "number" -%}
+        {{- "number" }}
+    {%- elif param_spec.type == "integer" -%}
+        {{- "number" }}
+    {%- elif param_spec.type == "boolean" -%}
+        {{- "boolean" }}
+
+    {%- elif param_spec.type == "object" -%}
+        {%- if param_spec.properties -%}
+            {{- "{\n" }}
+            {%- for prop_name, prop_spec in param_spec.properties.items() -%}
+                {{- prop_name -}}
+                {%- if prop_name not in (param_spec.required or []) -%}
+                    {{- "?" }}
+                {%- endif -%}
+                {{- ": " }}
+                {{ render_typescript_type(prop_spec, param_spec.required or []) }}
+                {%- if not loop.last -%}
+                    {{-", " }}
+                {%- endif -%}
+            {%- endfor -%}
+            {{- "}" }}
+        {%- else -%}
+            {{- "object" }}
+        {%- endif -%}
+    {%- else -%}
+        {{- "any" }}
+    {%- endif -%}
+{%- endmacro -%}
+
+{%- macro render_tool_namespace(namespace_name, tools) -%}
+    {{- "## " + namespace_name + "\n\n" }}
+    {{- "namespace " + namespace_name + " {\n\n" }}
+    {%- for tool in tools %}
+        {%- set tool = tool.function %}
+        {{- "// " + tool.description + "\n" }}
+        {{- "type "+ tool.name + " = " }}
+        {%- if tool.parameters and tool.parameters.properties %}
+            {{- "(_: {\n" }}
+            {%- for param_name, param_spec in tool.parameters.properties.items() %}
+                {%- if param_spec.description %}
+                    {{- "// " + param_spec.description + "\n" }}
+                {%- endif %}
+                {{- param_name }}
+                {%- if param_name not in (tool.parameters.required or []) -%}
+                    {{- "?" }}
+                {%- endif -%}
+                {{- ": " }}
+                {{- render_typescript_type(param_spec, tool.parameters.required or []) }}
+                {%- if param_spec.default is defined -%}
+                    {%- if param_spec.enum %}
+                        {{- ", // default: " + param_spec.default }}
+                    {%- elif param_spec.oneOf %}
+                        {{- "// default: " + param_spec.default }}
+                    {%- else %}
+                        {{- ", // default: " + param_spec.default|tojson }}
+                    {%- endif -%}
+                {%- endif -%}
+                {%- if not loop.last %}
+                    {{- ",\n" }}
+                {%- else %}
+                    {{- ",\n" }}
+                {%- endif -%}
+            {%- endfor %}
+            {{- "}) => any;\n\n" }}
+        {%- else -%}
+            {{- "() => any;\n\n" }}
+        {%- endif -%}
+    {%- endfor %}
+    {{- "} // namespace " + namespace_name }}
+{%- endmacro -%}
+
+{%- macro render_builtin_tools(browser_tool, python_tool) -%}
+    {%- if browser_tool %}
+        {{- "## browser\n\n" }}
+        {{- "// Tool for browsing.\n" }}
+        {{- "// The `cursor` appears in brackets before each browsing display: `[{cursor}]`.\n" }}
+        {{- "// Cite information from the tool using the following format:\n" }}
+        {{- "// `【{cursor}†L{line_start}(-L{line_end})?】`, for example: `【6†L9-L11】` or `【8†L3】`.\n" }}
+        {{- "// Do not quote more than 10 words directly from the tool output.\n" }}
+        {{- "// sources=web (default: web)\n" }}
+        {{- "namespace browser {\n\n" }}
+        {{- "// Searches for information related to `query` and displays `topn` results.\n" }}
+        {{- "type search = (_: {\n" }}
+        {{- "query: string,\n" }}
+        {{- "topn?: number, // default: 10\n" }}
+        {{- "source?: string,\n" }}
+        {{- "}) => any;\n\n" }}
+        {{- "// Opens the link `id` from the page indicated by `cursor` starting at line number `loc`, showing `num_lines` lines.\n" }}
+        {{- "// Valid link ids are displayed with the formatting: `【{id}†.*】`.\n" }}
+        {{- "// If `cursor` is not provided, the most recent page is implied.\n" }}
+        {{- "// If `id` is a string, it is treated as a fully qualified URL associated with `source`.\n" }}
+        {{- "// If `loc` is not provided, the viewport will be positioned at the beginning of the document or centered on the most relevant passage, if available.\n" }}
+        {{- "// Use this function without `id` to scroll to a new location of an opened page.\n" }}
+        {{- "type open = (_: {\n" }}
+        {{- "id?: number | string, // default: -1\n" }}
+        {{- "cursor?: number, // default: -1\n" }}
+        {{- "loc?: number, // default: -1\n" }}
+        {{- "num_lines?: number, // default: -1\n" }}
+        {{- "view_source?: boolean, // default: false\n" }}
+        {{- "source?: string,\n" }}
+        {{- "}) => any;\n\n" }}
+        {{- "// Finds exact matches of `pattern` in the current page, or the page given by `cursor`.\n" }}
+        {{- "type find = (_: {\n" }}
+        {{- "pattern: string,\n" }}
+        {{- "cursor?: number, // default: -1\n" }}
+        {{- "}) => any;\n\n" }}
+        {{- "} // namespace browser\n\n" }}
+    {%- endif -%}
+
+    {%- if python_tool %}
+        {{- "## python\n\n" }}
+        {{- "Use this tool to execute Python code in your chain of thought. The code will not be shown to the user. This tool should be used for internal reasoning, but not for code that is intended to be visible to the user (e.g. when creating plots, tables, or files).\n\n" }}
+        {{- "When you send a message containing Python code to python, it will be executed in a stateful Jupyter notebook environment. python will respond with the output of the execution or time out after 120.0 seconds. The drive at '/mnt/data' can be used to save and persist user files. Internet access for this session is UNKNOWN. Depends on the cluster.\n\n" }}
+    {%- endif -%}
+{%- endmacro -%}
+
+{#- System Message Construction ============================================ #}
+{%- macro build_system_message() -%}
+    {%- if model_identity is not defined %}
+        {%- set model_identity = "You are ChatGPT, a large language model trained by OpenAI." %}
+    {%- endif %}
+    {{- model_identity + "\n" }}
+    {{- "Knowledge cutoff: 2024-06\n" }}
+    {{- "Current date: " + strftime_now("%Y-%m-%d") + "\n\n" }}
+    {%- if reasoning_effort is not defined %}
+        {%- set reasoning_effort = "medium" %}
+    {%- endif %}
+    {{- "Reasoning: " + reasoning_effort + "\n\n" }}
+    {%- if builtin_tools %}
+        {{- "# Tools\n\n" }}
+        {%- set available_builtin_tools = namespace(browser=false, python=false) %}
+        {%- for tool in builtin_tools %}
+            {%- if tool == "browser" %}
+                {%- set available_builtin_tools.browser = true %}
+            {%- elif tool == "python" %}
+                {%- set available_builtin_tools.python = true %}
+            {%- endif %}
+        {%- endfor %}
+        {{- render_builtin_tools(available_builtin_tools.browser, available_builtin_tools.python) }}
+    {%- endif -%}
+    {{- "# Valid channels: analysis, commentary, final. Channel must be included for every message." }}
+    {%- if tools -%}
+        {{- "\nCalls to these tools must go to the commentary channel: 'functions'." }}
+    {%- endif -%}
+{%- endmacro -%}
+
+{#- Main Template Logic ================================================= #}
+{#- Set defaults #}
+
+{#- Render system message #}
+{{- "<|start|>system<|message|>" }}
+{{- build_system_message() }}
+{{- "<|end|>" }}
+
+{#- Extract developer message #}
+{%- if messages[0].role == "developer" or messages[0].role == "system" %}
+    {%- set developer_message = messages[0].content %}
+    {%- set loop_messages = messages[1:] %}
+{%- else %}
+    {%- set developer_message = "" %}
+    {%- set loop_messages = messages %}
+{%- endif %}
+
+{#- Render developer message #}
+{%- if developer_message or tools %}
+    {{- "<|start|>developer<|message|>" }}
+    {%- if developer_message %}
+        {{- "# Instructions\n\n" }}
+        {{- developer_message }}
+        {{- "\n\n" }}
+    {%- endif %}
+    {%- if tools -%}
+        {{- "# Tools\n\n" }}
+        {{- render_tool_namespace("functions", tools) }}
+    {%- endif -%}
+    {{- "<|end|>" }}
+{%- endif %}
+
+{#- Render messages #}
+{%- set last_tool_call = namespace(name=none) %}
+{%- for message in loop_messages -%}
+    {#- At this point only assistant/user/tool messages should remain #}
+    {%- if message.role == 'assistant' -%}
+        {#- Checks to ensure the messages are being passed in the format we expect #}
+        {%- if "content" in message %}
+            {%- if "<|channel|>analysis<|message|>" in message.content or "<|channel|>final<|message|>" in message.content %}
+                {{- raise_exception("You have passed a message containing <|channel|> tags in the content field. Instead of doing this, you should pass analysis messages (the string between '<|message|>' and '<|end|>') in the 'thinking' field, and final messages (the string between '<|message|>' and '<|end|>') in the 'content' field.") }}
+            {%- endif %}
+        {%- endif %}
+        {%- if "thinking" in message %}
+            {%- if "<|channel|>analysis<|message|>" in message.thinking or "<|channel|>final<|message|>" in message.thinking %}
+                {{- raise_exception("You have passed a message containing <|channel|> tags in the thinking field. Instead of doing this, you should pass analysis messages (the string between '<|message|>' and '<|end|>') in the 'thinking' field, and final messages (the string between '<|message|>' and '<|end|>') in the 'content' field.") }}
+            {%- endif %}
+        {%- endif %}
+        {%- if "tool_calls" in message %}
+            {#- We need very careful handling here - we want to drop the tool call analysis message if the model #}
+            {#- has output a later <|final|> message, but otherwise we want to retain it. This is the only case #}
+            {#- when we render CoT/analysis messages in inference. #}
+            {%- set future_final_message = namespace(found=false) %}
+            {%- for future_message in loop_messages[loop.index:] %}
+                {%- if future_message.role == 'assistant' and "tool_calls" not in future_message %}
+                    {%- set future_final_message.found = true %}
+                {%- endif %}
+            {%- endfor %}
+            {#- We assume max 1 tool call per message, and so we infer the tool call name #}
+            {#- in "tool" messages from the most recent assistant tool call name #}
+            {%- set tool_call = message.tool_calls[0] %}
+            {%- if tool_call.function %}
+                {%- set tool_call = tool_call.function %}
+            {%- endif %}
+            {%- if message.content and message.thinking %}
+                {{- raise_exception("Cannot pass both content and thinking in an assistant message with tool calls! Put the analysis message in one or the other, but not both.") }}
+            {%- elif message.content and not future_final_message.found %}
+                {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.content + "<|end|>" }}
+            {%- elif message.thinking and not future_final_message.found %}
+                {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.thinking + "<|end|>" }}
+            {%- endif %}
+            {{- "<|start|>assistant to=" }}
+            {{- "functions." + tool_call.name + "<|channel|>commentary " }}
+            {{- (tool_call.content_type if tool_call.content_type is defined else "json") + "<|message|>" }}
+            {{- tool_call.arguments|tojson }}
+            {{- "<|call|>" }}
+            {%- set last_tool_call.name = tool_call.name %}
+        {%- elif loop.last and not add_generation_prompt %}
+            {#- Only render the CoT if the final turn is an assistant turn and add_generation_prompt is false #}
+            {#- This is a situation that should only occur in training, never in inference. #}
+            {%- if "thinking" in message %}
+                {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.thinking + "<|end|>" }}
+            {%- endif %}
+            {#- <|return|> indicates the end of generation, but <|end|> does not #}
+            {#- <|return|> should never be an input to the model, but we include it as the final token #}
+            {#- when training, so the model learns to emit it. #}
+            {{- "<|start|>assistant<|channel|>final<|message|>" + message.content + "<|return|>" }}
+        {%- else %}
+            {#- CoT is dropped during all previous turns, so we never render it for inference #}
+            {{- "<|start|>assistant<|channel|>final<|message|>" + message.content + "<|end|>" }}
+            {%- set last_tool_call.name = none %}
+        {%- endif %}
+    {%- elif message.role == 'tool' -%}
+        {%- if last_tool_call.name is none %}
+            {{- raise_exception("Message has tool role, but there was no previous assistant message with a tool call!") }}
+        {%- endif %}
+        {{- "<|start|>functions." + last_tool_call.name }}
+        {{- " to=assistant<|channel|>commentary<|message|>" + message.content|tojson + "<|end|>" }}
+    {%- elif message.role == 'user' -%}
+        {{- "<|start|>user<|message|>" + message.content + "<|end|>" }}
+    {%- endif -%}
+{%- endfor -%}
+
+{#- Generation prompt #}
+{%- if add_generation_prompt -%}
+<|start|>assistant
+{%- endif -%}

tokenizer/tokenizer.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:928addc4a139e5382c06af49ebd495a4aa2aca99d13b872873e27bcbacc3ae4d
+size 27868440

tokenizer/tokenizer_config.json

@@ -0,0 +1,15 @@
+{
+  "backend": "tokenizers",
+  "bos_token": "<|startoftext|>",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|return|>",
+  "is_local": true,
+  "local_files_only": false,
+  "model_input_names": [
+    "input_ids",
+    "attention_mask"
+  ],
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<|endoftext|>",
+  "tokenizer_class": "TokenizersBackend"
+}

transformer/config.json

@@ -0,0 +1,474 @@
+{
+  "_class_name": "LensTransformer2DModel",
+  "_diffusers_version": "0.38.0",
+  "_name_or_path": "/workspace/.cache/huggingface/hub/models--microsoft--Lens-Turbo/snapshots/f21b81bed3bae7f93f16f5fd300af9c408f5816a/transformer",
+  "attention_head_dim": 64,
+  "axes_dims_rope": [
+    8,
+    28,
+    28
+  ],
+  "enc_hidden_dim": 2880,
+  "gate_mlp": true,
+  "in_channels": 128,
+  "inner_dim": 1536,
+  "multi_layer_encoder_feature": true,
+  "num_attention_heads": 24,
+  "num_layers": 48,
+  "out_channels": 32,
+  "patch_size": 2,
+  "quantization_config": {
+    "add_skip_keys": false,
+    "dequantize_fp32": false,
+    "dynamic_loss_threshold": null,
+    "group_size": 0,
+    "hadamard_group_size": 128,
+    "is_integer": true,
+    "is_training": false,
+    "modules_dtype_dict": {},
+    "modules_quant_config": {},
+    "modules_to_not_convert": [
+      "wte",
+      "norm",
+      ".img_out",
+      "multi_modal_projector",
+      ".norm_out",
+      "patch_embedding",
+      ".img_in",
+      ".context_embedder",
+      ".txt_out",
+      "patch_embed",
+      ".vid_in",
+      ".final_layer",
+      "time_text_embed",
+      "pos_embed",
+      "lm_head",
+      ".vid_out",
+      ".txt_in",
+      ".t_embedder",
+      ".x_embedder",
+      "patch_emb",
+      ".y_embedder",
+      ".condition_embedder",
+      ".proj_out",
+      ".time_embed",
+      ".emb_out",
+      ".emb_in",
+      "txt_norm.0.weight",
+      "txt_norm.1.weight",
+      "txt_norm.2.weight",
+      "txt_norm.3.weight",
+      "transformer_blocks.0.attn.norm_q.weight",
+      "transformer_blocks.0.attn.norm_k.weight",
+      "transformer_blocks.0.attn.norm_added_q.weight",
+      "transformer_blocks.0.attn.norm_added_k.weight",
+      "transformer_blocks.0.img_norm1.weight",
+      "transformer_blocks.0.img_norm2.weight",
+      "transformer_blocks.0.txt_norm1.weight",
+      "transformer_blocks.0.txt_norm2.weight",
+      "transformer_blocks.1.attn.norm_q.weight",
+      "transformer_blocks.1.attn.norm_k.weight",
+      "transformer_blocks.1.attn.norm_added_q.weight",
+      "transformer_blocks.1.attn.norm_added_k.weight",
+      "transformer_blocks.1.img_norm1.weight",
+      "transformer_blocks.1.img_norm2.weight",
+      "transformer_blocks.1.txt_norm1.weight",
+      "transformer_blocks.1.txt_norm2.weight",
+      "transformer_blocks.2.attn.norm_q.weight",
+      "transformer_blocks.2.attn.norm_k.weight",
+      "transformer_blocks.2.attn.norm_added_q.weight",
+      "transformer_blocks.2.attn.norm_added_k.weight",
+      "transformer_blocks.2.img_norm1.weight",
+      "transformer_blocks.2.img_norm2.weight",
+      "transformer_blocks.2.txt_norm1.weight",
+      "transformer_blocks.2.txt_norm2.weight",
+      "transformer_blocks.3.attn.norm_q.weight",
+      "transformer_blocks.3.attn.norm_k.weight",
+      "transformer_blocks.3.attn.norm_added_q.weight",
+      "transformer_blocks.3.attn.norm_added_k.weight",
+      "transformer_blocks.3.img_norm1.weight",
+      "transformer_blocks.3.img_norm2.weight",
+      "transformer_blocks.3.txt_norm1.weight",
+      "transformer_blocks.3.txt_norm2.weight",
+      "transformer_blocks.4.attn.norm_q.weight",
+      "transformer_blocks.4.attn.norm_k.weight",
+      "transformer_blocks.4.attn.norm_added_q.weight",
+      "transformer_blocks.4.attn.norm_added_k.weight",
+      "transformer_blocks.4.img_norm1.weight",
+      "transformer_blocks.4.img_norm2.weight",
+      "transformer_blocks.4.txt_norm1.weight",
+      "transformer_blocks.4.txt_norm2.weight",
+      "transformer_blocks.5.attn.norm_q.weight",
+      "transformer_blocks.5.attn.norm_k.weight",
+      "transformer_blocks.5.attn.norm_added_q.weight",
+      "transformer_blocks.5.attn.norm_added_k.weight",
+      "transformer_blocks.5.img_norm1.weight",
+      "transformer_blocks.5.img_norm2.weight",
+      "transformer_blocks.5.txt_norm1.weight",
+      "transformer_blocks.5.txt_norm2.weight",
+      "transformer_blocks.6.attn.norm_q.weight",
+      "transformer_blocks.6.attn.norm_k.weight",
+      "transformer_blocks.6.attn.norm_added_q.weight",
+      "transformer_blocks.6.attn.norm_added_k.weight",
+      "transformer_blocks.6.img_norm1.weight",
+      "transformer_blocks.6.img_norm2.weight",
+      "transformer_blocks.6.txt_norm1.weight",
+      "transformer_blocks.6.txt_norm2.weight",
+      "transformer_blocks.7.attn.norm_q.weight",
+      "transformer_blocks.7.attn.norm_k.weight",
+      "transformer_blocks.7.attn.norm_added_q.weight",
+      "transformer_blocks.7.attn.norm_added_k.weight",
+      "transformer_blocks.7.img_norm1.weight",
+      "transformer_blocks.7.img_norm2.weight",
+      "transformer_blocks.7.txt_norm1.weight",
+      "transformer_blocks.7.txt_norm2.weight",
+      "transformer_blocks.8.attn.norm_q.weight",
+      "transformer_blocks.8.attn.norm_k.weight",
+      "transformer_blocks.8.attn.norm_added_q.weight",
+      "transformer_blocks.8.attn.norm_added_k.weight",
+      "transformer_blocks.8.img_norm1.weight",
+      "transformer_blocks.8.img_norm2.weight",
+      "transformer_blocks.8.txt_norm1.weight",
+      "transformer_blocks.8.txt_norm2.weight",
+      "transformer_blocks.9.attn.norm_q.weight",
+      "transformer_blocks.9.attn.norm_k.weight",
+      "transformer_blocks.9.attn.norm_added_q.weight",
+      "transformer_blocks.9.attn.norm_added_k.weight",
+      "transformer_blocks.9.img_norm1.weight",
+      "transformer_blocks.9.img_norm2.weight",
+      "transformer_blocks.9.txt_norm1.weight",
+      "transformer_blocks.9.txt_norm2.weight",
+      "transformer_blocks.10.attn.norm_q.weight",
+      "transformer_blocks.10.attn.norm_k.weight",
+      "transformer_blocks.10.attn.norm_added_q.weight",
+      "transformer_blocks.10.attn.norm_added_k.weight",
+      "transformer_blocks.10.img_norm1.weight",
+      "transformer_blocks.10.img_norm2.weight",
+      "transformer_blocks.10.txt_norm1.weight",
+      "transformer_blocks.10.txt_norm2.weight",
+      "transformer_blocks.11.attn.norm_q.weight",
+      "transformer_blocks.11.attn.norm_k.weight",
+      "transformer_blocks.11.attn.norm_added_q.weight",
+      "transformer_blocks.11.attn.norm_added_k.weight",
+      "transformer_blocks.11.img_norm1.weight",
+      "transformer_blocks.11.img_norm2.weight",
+      "transformer_blocks.11.txt_norm1.weight",
+      "transformer_blocks.11.txt_norm2.weight",
+      "transformer_blocks.12.attn.norm_q.weight",
+      "transformer_blocks.12.attn.norm_k.weight",
+      "transformer_blocks.12.attn.norm_added_q.weight",
+      "transformer_blocks.12.attn.norm_added_k.weight",
+      "transformer_blocks.12.img_norm1.weight",
+      "transformer_blocks.12.img_norm2.weight",
+      "transformer_blocks.12.txt_norm1.weight",
+      "transformer_blocks.12.txt_norm2.weight",
+      "transformer_blocks.13.attn.norm_q.weight",
+      "transformer_blocks.13.attn.norm_k.weight",
+      "transformer_blocks.13.attn.norm_added_q.weight",
+      "transformer_blocks.13.attn.norm_added_k.weight",
+      "transformer_blocks.13.img_norm1.weight",
+      "transformer_blocks.13.img_norm2.weight",
+      "transformer_blocks.13.txt_norm1.weight",
+      "transformer_blocks.13.txt_norm2.weight",
+      "transformer_blocks.14.attn.norm_q.weight",
+      "transformer_blocks.14.attn.norm_k.weight",
+      "transformer_blocks.14.attn.norm_added_q.weight",
+      "transformer_blocks.14.attn.norm_added_k.weight",
+      "transformer_blocks.14.img_norm1.weight",
+      "transformer_blocks.14.img_norm2.weight",
+      "transformer_blocks.14.txt_norm1.weight",
+      "transformer_blocks.14.txt_norm2.weight",
+      "transformer_blocks.15.attn.norm_q.weight",
+      "transformer_blocks.15.attn.norm_k.weight",
+      "transformer_blocks.15.attn.norm_added_q.weight",
+      "transformer_blocks.15.attn.norm_added_k.weight",
+      "transformer_blocks.15.img_norm1.weight",
+      "transformer_blocks.15.img_norm2.weight",
+      "transformer_blocks.15.txt_norm1.weight",
+      "transformer_blocks.15.txt_norm2.weight",
+      "transformer_blocks.16.attn.norm_q.weight",
+      "transformer_blocks.16.attn.norm_k.weight",
+      "transformer_blocks.16.attn.norm_added_q.weight",
+      "transformer_blocks.16.attn.norm_added_k.weight",
+      "transformer_blocks.16.img_norm1.weight",
+      "transformer_blocks.16.img_norm2.weight",
+      "transformer_blocks.16.txt_norm1.weight",
+      "transformer_blocks.16.txt_norm2.weight",
+      "transformer_blocks.17.attn.norm_q.weight",
+      "transformer_blocks.17.attn.norm_k.weight",
+      "transformer_blocks.17.attn.norm_added_q.weight",
+      "transformer_blocks.17.attn.norm_added_k.weight",
+      "transformer_blocks.17.img_norm1.weight",
+      "transformer_blocks.17.img_norm2.weight",
+      "transformer_blocks.17.txt_norm1.weight",
+      "transformer_blocks.17.txt_norm2.weight",
+      "transformer_blocks.18.attn.norm_q.weight",
+      "transformer_blocks.18.attn.norm_k.weight",
+      "transformer_blocks.18.attn.norm_added_q.weight",
+      "transformer_blocks.18.attn.norm_added_k.weight",
+      "transformer_blocks.18.img_norm1.weight",
+      "transformer_blocks.18.img_norm2.weight",
+      "transformer_blocks.18.txt_norm1.weight",
+      "transformer_blocks.18.txt_norm2.weight",
+      "transformer_blocks.19.attn.norm_q.weight",
+      "transformer_blocks.19.attn.norm_k.weight",
+      "transformer_blocks.19.attn.norm_added_q.weight",
+      "transformer_blocks.19.attn.norm_added_k.weight",
+      "transformer_blocks.19.img_norm1.weight",
+      "transformer_blocks.19.img_norm2.weight",
+      "transformer_blocks.19.txt_norm1.weight",
+      "transformer_blocks.19.txt_norm2.weight",
+      "transformer_blocks.20.attn.norm_q.weight",
+      "transformer_blocks.20.attn.norm_k.weight",
+      "transformer_blocks.20.attn.norm_added_q.weight",
+      "transformer_blocks.20.attn.norm_added_k.weight",
+      "transformer_blocks.20.img_norm1.weight",
+      "transformer_blocks.20.img_norm2.weight",
+      "transformer_blocks.20.txt_norm1.weight",
+      "transformer_blocks.20.txt_norm2.weight",
+      "transformer_blocks.21.attn.norm_q.weight",
+      "transformer_blocks.21.attn.norm_k.weight",
+      "transformer_blocks.21.attn.norm_added_q.weight",
+      "transformer_blocks.21.attn.norm_added_k.weight",
+      "transformer_blocks.21.img_norm1.weight",
+      "transformer_blocks.21.img_norm2.weight",
+      "transformer_blocks.21.txt_norm1.weight",
+      "transformer_blocks.21.txt_norm2.weight",
+      "transformer_blocks.22.attn.norm_q.weight",
+      "transformer_blocks.22.attn.norm_k.weight",
+      "transformer_blocks.22.attn.norm_added_q.weight",
+      "transformer_blocks.22.attn.norm_added_k.weight",
+      "transformer_blocks.22.img_norm1.weight",
+      "transformer_blocks.22.img_norm2.weight",
+      "transformer_blocks.22.txt_norm1.weight",
+      "transformer_blocks.22.txt_norm2.weight",
+      "transformer_blocks.23.attn.norm_q.weight",
+      "transformer_blocks.23.attn.norm_k.weight",
+      "transformer_blocks.23.attn.norm_added_q.weight",
+      "transformer_blocks.23.attn.norm_added_k.weight",
+      "transformer_blocks.23.img_norm1.weight",
+      "transformer_blocks.23.img_norm2.weight",
+      "transformer_blocks.23.txt_norm1.weight",
+      "transformer_blocks.23.txt_norm2.weight",
+      "transformer_blocks.24.attn.norm_q.weight",
+      "transformer_blocks.24.attn.norm_k.weight",
+      "transformer_blocks.24.attn.norm_added_q.weight",
+      "transformer_blocks.24.attn.norm_added_k.weight",
+      "transformer_blocks.24.img_norm1.weight",
+      "transformer_blocks.24.img_norm2.weight",
+      "transformer_blocks.24.txt_norm1.weight",
+      "transformer_blocks.24.txt_norm2.weight",
+      "transformer_blocks.25.attn.norm_q.weight",
+      "transformer_blocks.25.attn.norm_k.weight",
+      "transformer_blocks.25.attn.norm_added_q.weight",
+      "transformer_blocks.25.attn.norm_added_k.weight",
+      "transformer_blocks.25.img_norm1.weight",
+      "transformer_blocks.25.img_norm2.weight",
+      "transformer_blocks.25.txt_norm1.weight",
+      "transformer_blocks.25.txt_norm2.weight",
+      "transformer_blocks.26.attn.norm_q.weight",
+      "transformer_blocks.26.attn.norm_k.weight",
+      "transformer_blocks.26.attn.norm_added_q.weight",
+      "transformer_blocks.26.attn.norm_added_k.weight",
+      "transformer_blocks.26.img_norm1.weight",
+      "transformer_blocks.26.img_norm2.weight",
+      "transformer_blocks.26.txt_norm1.weight",
+      "transformer_blocks.26.txt_norm2.weight",
+      "transformer_blocks.27.attn.norm_q.weight",
+      "transformer_blocks.27.attn.norm_k.weight",
+      "transformer_blocks.27.attn.norm_added_q.weight",
+      "transformer_blocks.27.attn.norm_added_k.weight",
+      "transformer_blocks.27.img_norm1.weight",
+      "transformer_blocks.27.img_norm2.weight",
+      "transformer_blocks.27.txt_norm1.weight",
+      "transformer_blocks.27.txt_norm2.weight",
+      "transformer_blocks.28.attn.norm_q.weight",
+      "transformer_blocks.28.attn.norm_k.weight",
+      "transformer_blocks.28.attn.norm_added_q.weight",
+      "transformer_blocks.28.attn.norm_added_k.weight",
+      "transformer_blocks.28.img_norm1.weight",
+      "transformer_blocks.28.img_norm2.weight",
+      "transformer_blocks.28.txt_norm1.weight",
+      "transformer_blocks.28.txt_norm2.weight",
+      "transformer_blocks.29.attn.norm_q.weight",
+      "transformer_blocks.29.attn.norm_k.weight",
+      "transformer_blocks.29.attn.norm_added_q.weight",
+      "transformer_blocks.29.attn.norm_added_k.weight",
+      "transformer_blocks.29.img_norm1.weight",
+      "transformer_blocks.29.img_norm2.weight",
+      "transformer_blocks.29.txt_norm1.weight",
+      "transformer_blocks.29.txt_norm2.weight",
+      "transformer_blocks.30.attn.norm_q.weight",
+      "transformer_blocks.30.attn.norm_k.weight",
+      "transformer_blocks.30.attn.norm_added_q.weight",
+      "transformer_blocks.30.attn.norm_added_k.weight",
+      "transformer_blocks.30.img_norm1.weight",
+      "transformer_blocks.30.img_norm2.weight",
+      "transformer_blocks.30.txt_norm1.weight",
+      "transformer_blocks.30.txt_norm2.weight",
+      "transformer_blocks.31.attn.norm_q.weight",
+      "transformer_blocks.31.attn.norm_k.weight",
+      "transformer_blocks.31.attn.norm_added_q.weight",
+      "transformer_blocks.31.attn.norm_added_k.weight",
+      "transformer_blocks.31.img_norm1.weight",
+      "transformer_blocks.31.img_norm2.weight",
+      "transformer_blocks.31.txt_norm1.weight",
+      "transformer_blocks.31.txt_norm2.weight",
+      "transformer_blocks.32.attn.norm_q.weight",
+      "transformer_blocks.32.attn.norm_k.weight",
+      "transformer_blocks.32.attn.norm_added_q.weight",
+      "transformer_blocks.32.attn.norm_added_k.weight",
+      "transformer_blocks.32.img_norm1.weight",
+      "transformer_blocks.32.img_norm2.weight",
+      "transformer_blocks.32.txt_norm1.weight",
+      "transformer_blocks.32.txt_norm2.weight",
+      "transformer_blocks.33.attn.norm_q.weight",
+      "transformer_blocks.33.attn.norm_k.weight",
+      "transformer_blocks.33.attn.norm_added_q.weight",
+      "transformer_blocks.33.attn.norm_added_k.weight",
+      "transformer_blocks.33.img_norm1.weight",
+      "transformer_blocks.33.img_norm2.weight",
+      "transformer_blocks.33.txt_norm1.weight",
+      "transformer_blocks.33.txt_norm2.weight",
+      "transformer_blocks.34.attn.norm_q.weight",
+      "transformer_blocks.34.attn.norm_k.weight",
+      "transformer_blocks.34.attn.norm_added_q.weight",
+      "transformer_blocks.34.attn.norm_added_k.weight",
+      "transformer_blocks.34.img_norm1.weight",
+      "transformer_blocks.34.img_norm2.weight",
+      "transformer_blocks.34.txt_norm1.weight",
+      "transformer_blocks.34.txt_norm2.weight",
+      "transformer_blocks.35.attn.norm_q.weight",
+      "transformer_blocks.35.attn.norm_k.weight",
+      "transformer_blocks.35.attn.norm_added_q.weight",
+      "transformer_blocks.35.attn.norm_added_k.weight",
+      "transformer_blocks.35.img_norm1.weight",
+      "transformer_blocks.35.img_norm2.weight",
+      "transformer_blocks.35.txt_norm1.weight",
+      "transformer_blocks.35.txt_norm2.weight",
+      "transformer_blocks.36.attn.norm_q.weight",
+      "transformer_blocks.36.attn.norm_k.weight",
+      "transformer_blocks.36.attn.norm_added_q.weight",
+      "transformer_blocks.36.attn.norm_added_k.weight",
+      "transformer_blocks.36.img_norm1.weight",
+      "transformer_blocks.36.img_norm2.weight",
+      "transformer_blocks.36.txt_norm1.weight",
+      "transformer_blocks.36.txt_norm2.weight",
+      "transformer_blocks.37.attn.norm_q.weight",
+      "transformer_blocks.37.attn.norm_k.weight",
+      "transformer_blocks.37.attn.norm_added_q.weight",
+      "transformer_blocks.37.attn.norm_added_k.weight",
+      "transformer_blocks.37.img_norm1.weight",
+      "transformer_blocks.37.img_norm2.weight",
+      "transformer_blocks.37.txt_norm1.weight",
+      "transformer_blocks.37.txt_norm2.weight",
+      "transformer_blocks.38.attn.norm_q.weight",
+      "transformer_blocks.38.attn.norm_k.weight",
+      "transformer_blocks.38.attn.norm_added_q.weight",
+      "transformer_blocks.38.attn.norm_added_k.weight",
+      "transformer_blocks.38.img_norm1.weight",
+      "transformer_blocks.38.img_norm2.weight",
+      "transformer_blocks.38.txt_norm1.weight",
+      "transformer_blocks.38.txt_norm2.weight",
+      "transformer_blocks.39.attn.norm_q.weight",
+      "transformer_blocks.39.attn.norm_k.weight",
+      "transformer_blocks.39.attn.norm_added_q.weight",
+      "transformer_blocks.39.attn.norm_added_k.weight",
+      "transformer_blocks.39.img_norm1.weight",
+      "transformer_blocks.39.img_norm2.weight",
+      "transformer_blocks.39.txt_norm1.weight",
+      "transformer_blocks.39.txt_norm2.weight",
+      "transformer_blocks.40.attn.norm_q.weight",
+      "transformer_blocks.40.attn.norm_k.weight",
+      "transformer_blocks.40.attn.norm_added_q.weight",
+      "transformer_blocks.40.attn.norm_added_k.weight",
+      "transformer_blocks.40.img_norm1.weight",
+      "transformer_blocks.40.img_norm2.weight",
+      "transformer_blocks.40.txt_norm1.weight",
+      "transformer_blocks.40.txt_norm2.weight",
+      "transformer_blocks.41.attn.norm_q.weight",
+      "transformer_blocks.41.attn.norm_k.weight",
+      "transformer_blocks.41.attn.norm_added_q.weight",
+      "transformer_blocks.41.attn.norm_added_k.weight",
+      "transformer_blocks.41.img_norm1.weight",
+      "transformer_blocks.41.img_norm2.weight",
+      "transformer_blocks.41.txt_norm1.weight",
+      "transformer_blocks.41.txt_norm2.weight",
+      "transformer_blocks.42.attn.norm_q.weight",
+      "transformer_blocks.42.attn.norm_k.weight",
+      "transformer_blocks.42.attn.norm_added_q.weight",
+      "transformer_blocks.42.attn.norm_added_k.weight",
+      "transformer_blocks.42.img_norm1.weight",
+      "transformer_blocks.42.img_norm2.weight",
+      "transformer_blocks.42.txt_norm1.weight",
+      "transformer_blocks.42.txt_norm2.weight",
+      "transformer_blocks.43.attn.norm_q.weight",
+      "transformer_blocks.43.attn.norm_k.weight",
+      "transformer_blocks.43.attn.norm_added_q.weight",
+      "transformer_blocks.43.attn.norm_added_k.weight",
+      "transformer_blocks.43.img_norm1.weight",
+      "transformer_blocks.43.img_norm2.weight",
+      "transformer_blocks.43.txt_norm1.weight",
+      "transformer_blocks.43.txt_norm2.weight",
+      "transformer_blocks.44.attn.norm_q.weight",
+      "transformer_blocks.44.attn.norm_k.weight",
+      "transformer_blocks.44.attn.norm_added_q.weight",
+      "transformer_blocks.44.attn.norm_added_k.weight",
+      "transformer_blocks.44.img_norm1.weight",
+      "transformer_blocks.44.img_norm2.weight",
+      "transformer_blocks.44.txt_norm1.weight",
+      "transformer_blocks.44.txt_norm2.weight",
+      "transformer_blocks.45.attn.norm_q.weight",
+      "transformer_blocks.45.attn.norm_k.weight",
+      "transformer_blocks.45.attn.norm_added_q.weight",
+      "transformer_blocks.45.attn.norm_added_k.weight",
+      "transformer_blocks.45.img_norm1.weight",
+      "transformer_blocks.45.img_norm2.weight",
+      "transformer_blocks.45.txt_norm1.weight",
+      "transformer_blocks.45.txt_norm2.weight",
+      "transformer_blocks.46.attn.norm_q.weight",
+      "transformer_blocks.46.attn.norm_k.weight",
+      "transformer_blocks.46.attn.norm_added_q.weight",
+      "transformer_blocks.46.attn.norm_added_k.weight",
+      "transformer_blocks.46.img_norm1.weight",
+      "transformer_blocks.46.img_norm2.weight",
+      "transformer_blocks.46.txt_norm1.weight",
+      "transformer_blocks.46.txt_norm2.weight",
+      "transformer_blocks.47.attn.norm_q.weight",
+      "transformer_blocks.47.attn.norm_k.weight",
+      "transformer_blocks.47.attn.norm_added_q.weight",
+      "transformer_blocks.47.attn.norm_added_k.weight",
+      "transformer_blocks.47.img_norm1.weight",
+      "transformer_blocks.47.img_norm2.weight",
+      "transformer_blocks.47.txt_norm1.weight",
+      "transformer_blocks.47.txt_norm2.weight"
+    ],
+    "modules_to_not_use_matmul": [],
+    "non_blocking": false,
+    "quant_conv": false,
+    "quant_embedding": false,
+    "quant_method": "sdnq",
+    "quantization_device": null,
+    "quantized_matmul_dtype": "int8",
+    "return_device": null,
+    "sdnq_version": "0.2.0",
+    "svd_rank": 32,
+    "svd_steps": 8,
+    "use_dynamic_quantization": false,
+    "use_grad_ckpt": true,
+    "use_hadamard": false,
+    "use_quantized_matmul": true,
+    "use_quantized_matmul_conv": false,
+    "use_static_quantization": true,
+    "use_stochastic_rounding": false,
+    "use_svd": false,
+    "weights_dtype": "uint4"
+  },
+  "rms_norm": true,
+  "selected_layer_index": [
+    5,
+    11,
+    17,
+    23
+  ]
+}

transformer/diffusion_pytorch_model.safetensors

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

transformer/quantization_config.json

@@ -0,0 +1,447 @@
+{
+  "add_skip_keys": false,
+  "dequantize_fp32": false,
+  "dynamic_loss_threshold": null,
+  "group_size": 0,
+  "hadamard_group_size": 128,
+  "is_integer": true,
+  "is_training": false,
+  "modules_dtype_dict": {},
+  "modules_quant_config": {},
+  "modules_to_not_convert": [
+    "wte",
+    "norm",
+    ".img_out",
+    "multi_modal_projector",
+    ".norm_out",
+    "patch_embedding",
+    ".img_in",
+    ".context_embedder",
+    ".txt_out",
+    "patch_embed",
+    ".vid_in",
+    ".final_layer",
+    "time_text_embed",
+    "pos_embed",
+    "lm_head",
+    ".vid_out",
+    ".txt_in",
+    ".t_embedder",
+    ".x_embedder",
+    "patch_emb",
+    ".y_embedder",
+    ".condition_embedder",
+    ".proj_out",
+    ".time_embed",
+    ".emb_out",
+    ".emb_in",
+    "txt_norm.0.weight",
+    "txt_norm.1.weight",
+    "txt_norm.2.weight",
+    "txt_norm.3.weight",
+    "transformer_blocks.0.attn.norm_q.weight",
+    "transformer_blocks.0.attn.norm_k.weight",
+    "transformer_blocks.0.attn.norm_added_q.weight",
+    "transformer_blocks.0.attn.norm_added_k.weight",
+    "transformer_blocks.0.img_norm1.weight",
+    "transformer_blocks.0.img_norm2.weight",
+    "transformer_blocks.0.txt_norm1.weight",
+    "transformer_blocks.0.txt_norm2.weight",
+    "transformer_blocks.1.attn.norm_q.weight",
+    "transformer_blocks.1.attn.norm_k.weight",
+    "transformer_blocks.1.attn.norm_added_q.weight",
+    "transformer_blocks.1.attn.norm_added_k.weight",
+    "transformer_blocks.1.img_norm1.weight",
+    "transformer_blocks.1.img_norm2.weight",
+    "transformer_blocks.1.txt_norm1.weight",
+    "transformer_blocks.1.txt_norm2.weight",
+    "transformer_blocks.2.attn.norm_q.weight",
+    "transformer_blocks.2.attn.norm_k.weight",
+    "transformer_blocks.2.attn.norm_added_q.weight",
+    "transformer_blocks.2.attn.norm_added_k.weight",
+    "transformer_blocks.2.img_norm1.weight",
+    "transformer_blocks.2.img_norm2.weight",
+    "transformer_blocks.2.txt_norm1.weight",
+    "transformer_blocks.2.txt_norm2.weight",
+    "transformer_blocks.3.attn.norm_q.weight",
+    "transformer_blocks.3.attn.norm_k.weight",
+    "transformer_blocks.3.attn.norm_added_q.weight",
+    "transformer_blocks.3.attn.norm_added_k.weight",
+    "transformer_blocks.3.img_norm1.weight",
+    "transformer_blocks.3.img_norm2.weight",
+    "transformer_blocks.3.txt_norm1.weight",
+    "transformer_blocks.3.txt_norm2.weight",
+    "transformer_blocks.4.attn.norm_q.weight",
+    "transformer_blocks.4.attn.norm_k.weight",
+    "transformer_blocks.4.attn.norm_added_q.weight",
+    "transformer_blocks.4.attn.norm_added_k.weight",
+    "transformer_blocks.4.img_norm1.weight",
+    "transformer_blocks.4.img_norm2.weight",
+    "transformer_blocks.4.txt_norm1.weight",
+    "transformer_blocks.4.txt_norm2.weight",
+    "transformer_blocks.5.attn.norm_q.weight",
+    "transformer_blocks.5.attn.norm_k.weight",
+    "transformer_blocks.5.attn.norm_added_q.weight",
+    "transformer_blocks.5.attn.norm_added_k.weight",
+    "transformer_blocks.5.img_norm1.weight",
+    "transformer_blocks.5.img_norm2.weight",
+    "transformer_blocks.5.txt_norm1.weight",
+    "transformer_blocks.5.txt_norm2.weight",
+    "transformer_blocks.6.attn.norm_q.weight",
+    "transformer_blocks.6.attn.norm_k.weight",
+    "transformer_blocks.6.attn.norm_added_q.weight",
+    "transformer_blocks.6.attn.norm_added_k.weight",
+    "transformer_blocks.6.img_norm1.weight",
+    "transformer_blocks.6.img_norm2.weight",
+    "transformer_blocks.6.txt_norm1.weight",
+    "transformer_blocks.6.txt_norm2.weight",
+    "transformer_blocks.7.attn.norm_q.weight",
+    "transformer_blocks.7.attn.norm_k.weight",
+    "transformer_blocks.7.attn.norm_added_q.weight",
+    "transformer_blocks.7.attn.norm_added_k.weight",
+    "transformer_blocks.7.img_norm1.weight",
+    "transformer_blocks.7.img_norm2.weight",
+    "transformer_blocks.7.txt_norm1.weight",
+    "transformer_blocks.7.txt_norm2.weight",
+    "transformer_blocks.8.attn.norm_q.weight",
+    "transformer_blocks.8.attn.norm_k.weight",
+    "transformer_blocks.8.attn.norm_added_q.weight",
+    "transformer_blocks.8.attn.norm_added_k.weight",
+    "transformer_blocks.8.img_norm1.weight",
+    "transformer_blocks.8.img_norm2.weight",
+    "transformer_blocks.8.txt_norm1.weight",
+    "transformer_blocks.8.txt_norm2.weight",
+    "transformer_blocks.9.attn.norm_q.weight",
+    "transformer_blocks.9.attn.norm_k.weight",
+    "transformer_blocks.9.attn.norm_added_q.weight",
+    "transformer_blocks.9.attn.norm_added_k.weight",
+    "transformer_blocks.9.img_norm1.weight",
+    "transformer_blocks.9.img_norm2.weight",
+    "transformer_blocks.9.txt_norm1.weight",
+    "transformer_blocks.9.txt_norm2.weight",
+    "transformer_blocks.10.attn.norm_q.weight",
+    "transformer_blocks.10.attn.norm_k.weight",
+    "transformer_blocks.10.attn.norm_added_q.weight",
+    "transformer_blocks.10.attn.norm_added_k.weight",
+    "transformer_blocks.10.img_norm1.weight",
+    "transformer_blocks.10.img_norm2.weight",
+    "transformer_blocks.10.txt_norm1.weight",
+    "transformer_blocks.10.txt_norm2.weight",
+    "transformer_blocks.11.attn.norm_q.weight",
+    "transformer_blocks.11.attn.norm_k.weight",
+    "transformer_blocks.11.attn.norm_added_q.weight",
+    "transformer_blocks.11.attn.norm_added_k.weight",
+    "transformer_blocks.11.img_norm1.weight",
+    "transformer_blocks.11.img_norm2.weight",
+    "transformer_blocks.11.txt_norm1.weight",
+    "transformer_blocks.11.txt_norm2.weight",
+    "transformer_blocks.12.attn.norm_q.weight",
+    "transformer_blocks.12.attn.norm_k.weight",
+    "transformer_blocks.12.attn.norm_added_q.weight",
+    "transformer_blocks.12.attn.norm_added_k.weight",
+    "transformer_blocks.12.img_norm1.weight",
+    "transformer_blocks.12.img_norm2.weight",
+    "transformer_blocks.12.txt_norm1.weight",
+    "transformer_blocks.12.txt_norm2.weight",
+    "transformer_blocks.13.attn.norm_q.weight",
+    "transformer_blocks.13.attn.norm_k.weight",
+    "transformer_blocks.13.attn.norm_added_q.weight",
+    "transformer_blocks.13.attn.norm_added_k.weight",
+    "transformer_blocks.13.img_norm1.weight",
+    "transformer_blocks.13.img_norm2.weight",
+    "transformer_blocks.13.txt_norm1.weight",
+    "transformer_blocks.13.txt_norm2.weight",
+    "transformer_blocks.14.attn.norm_q.weight",
+    "transformer_blocks.14.attn.norm_k.weight",
+    "transformer_blocks.14.attn.norm_added_q.weight",
+    "transformer_blocks.14.attn.norm_added_k.weight",
+    "transformer_blocks.14.img_norm1.weight",
+    "transformer_blocks.14.img_norm2.weight",
+    "transformer_blocks.14.txt_norm1.weight",
+    "transformer_blocks.14.txt_norm2.weight",
+    "transformer_blocks.15.attn.norm_q.weight",
+    "transformer_blocks.15.attn.norm_k.weight",
+    "transformer_blocks.15.attn.norm_added_q.weight",
+    "transformer_blocks.15.attn.norm_added_k.weight",
+    "transformer_blocks.15.img_norm1.weight",
+    "transformer_blocks.15.img_norm2.weight",
+    "transformer_blocks.15.txt_norm1.weight",
+    "transformer_blocks.15.txt_norm2.weight",
+    "transformer_blocks.16.attn.norm_q.weight",
+    "transformer_blocks.16.attn.norm_k.weight",
+    "transformer_blocks.16.attn.norm_added_q.weight",
+    "transformer_blocks.16.attn.norm_added_k.weight",
+    "transformer_blocks.16.img_norm1.weight",
+    "transformer_blocks.16.img_norm2.weight",
+    "transformer_blocks.16.txt_norm1.weight",
+    "transformer_blocks.16.txt_norm2.weight",
+    "transformer_blocks.17.attn.norm_q.weight",
+    "transformer_blocks.17.attn.norm_k.weight",
+    "transformer_blocks.17.attn.norm_added_q.weight",
+    "transformer_blocks.17.attn.norm_added_k.weight",
+    "transformer_blocks.17.img_norm1.weight",
+    "transformer_blocks.17.img_norm2.weight",
+    "transformer_blocks.17.txt_norm1.weight",
+    "transformer_blocks.17.txt_norm2.weight",
+    "transformer_blocks.18.attn.norm_q.weight",
+    "transformer_blocks.18.attn.norm_k.weight",
+    "transformer_blocks.18.attn.norm_added_q.weight",
+    "transformer_blocks.18.attn.norm_added_k.weight",
+    "transformer_blocks.18.img_norm1.weight",
+    "transformer_blocks.18.img_norm2.weight",
+    "transformer_blocks.18.txt_norm1.weight",
+    "transformer_blocks.18.txt_norm2.weight",
+    "transformer_blocks.19.attn.norm_q.weight",
+    "transformer_blocks.19.attn.norm_k.weight",
+    "transformer_blocks.19.attn.norm_added_q.weight",
+    "transformer_blocks.19.attn.norm_added_k.weight",
+    "transformer_blocks.19.img_norm1.weight",
+    "transformer_blocks.19.img_norm2.weight",
+    "transformer_blocks.19.txt_norm1.weight",
+    "transformer_blocks.19.txt_norm2.weight",
+    "transformer_blocks.20.attn.norm_q.weight",
+    "transformer_blocks.20.attn.norm_k.weight",
+    "transformer_blocks.20.attn.norm_added_q.weight",
+    "transformer_blocks.20.attn.norm_added_k.weight",
+    "transformer_blocks.20.img_norm1.weight",
+    "transformer_blocks.20.img_norm2.weight",
+    "transformer_blocks.20.txt_norm1.weight",
+    "transformer_blocks.20.txt_norm2.weight",
+    "transformer_blocks.21.attn.norm_q.weight",
+    "transformer_blocks.21.attn.norm_k.weight",
+    "transformer_blocks.21.attn.norm_added_q.weight",
+    "transformer_blocks.21.attn.norm_added_k.weight",
+    "transformer_blocks.21.img_norm1.weight",
+    "transformer_blocks.21.img_norm2.weight",
+    "transformer_blocks.21.txt_norm1.weight",
+    "transformer_blocks.21.txt_norm2.weight",
+    "transformer_blocks.22.attn.norm_q.weight",
+    "transformer_blocks.22.attn.norm_k.weight",
+    "transformer_blocks.22.attn.norm_added_q.weight",
+    "transformer_blocks.22.attn.norm_added_k.weight",
+    "transformer_blocks.22.img_norm1.weight",
+    "transformer_blocks.22.img_norm2.weight",
+    "transformer_blocks.22.txt_norm1.weight",
+    "transformer_blocks.22.txt_norm2.weight",
+    "transformer_blocks.23.attn.norm_q.weight",
+    "transformer_blocks.23.attn.norm_k.weight",
+    "transformer_blocks.23.attn.norm_added_q.weight",
+    "transformer_blocks.23.attn.norm_added_k.weight",
+    "transformer_blocks.23.img_norm1.weight",
+    "transformer_blocks.23.img_norm2.weight",
+    "transformer_blocks.23.txt_norm1.weight",
+    "transformer_blocks.23.txt_norm2.weight",
+    "transformer_blocks.24.attn.norm_q.weight",
+    "transformer_blocks.24.attn.norm_k.weight",
+    "transformer_blocks.24.attn.norm_added_q.weight",
+    "transformer_blocks.24.attn.norm_added_k.weight",
+    "transformer_blocks.24.img_norm1.weight",
+    "transformer_blocks.24.img_norm2.weight",
+    "transformer_blocks.24.txt_norm1.weight",
+    "transformer_blocks.24.txt_norm2.weight",
+    "transformer_blocks.25.attn.norm_q.weight",
+    "transformer_blocks.25.attn.norm_k.weight",
+    "transformer_blocks.25.attn.norm_added_q.weight",
+    "transformer_blocks.25.attn.norm_added_k.weight",
+    "transformer_blocks.25.img_norm1.weight",
+    "transformer_blocks.25.img_norm2.weight",
+    "transformer_blocks.25.txt_norm1.weight",
+    "transformer_blocks.25.txt_norm2.weight",
+    "transformer_blocks.26.attn.norm_q.weight",
+    "transformer_blocks.26.attn.norm_k.weight",
+    "transformer_blocks.26.attn.norm_added_q.weight",
+    "transformer_blocks.26.attn.norm_added_k.weight",
+    "transformer_blocks.26.img_norm1.weight",
+    "transformer_blocks.26.img_norm2.weight",
+    "transformer_blocks.26.txt_norm1.weight",
+    "transformer_blocks.26.txt_norm2.weight",
+    "transformer_blocks.27.attn.norm_q.weight",
+    "transformer_blocks.27.attn.norm_k.weight",
+    "transformer_blocks.27.attn.norm_added_q.weight",
+    "transformer_blocks.27.attn.norm_added_k.weight",
+    "transformer_blocks.27.img_norm1.weight",
+    "transformer_blocks.27.img_norm2.weight",
+    "transformer_blocks.27.txt_norm1.weight",
+    "transformer_blocks.27.txt_norm2.weight",
+    "transformer_blocks.28.attn.norm_q.weight",
+    "transformer_blocks.28.attn.norm_k.weight",
+    "transformer_blocks.28.attn.norm_added_q.weight",
+    "transformer_blocks.28.attn.norm_added_k.weight",
+    "transformer_blocks.28.img_norm1.weight",
+    "transformer_blocks.28.img_norm2.weight",
+    "transformer_blocks.28.txt_norm1.weight",
+    "transformer_blocks.28.txt_norm2.weight",
+    "transformer_blocks.29.attn.norm_q.weight",
+    "transformer_blocks.29.attn.norm_k.weight",
+    "transformer_blocks.29.attn.norm_added_q.weight",
+    "transformer_blocks.29.attn.norm_added_k.weight",
+    "transformer_blocks.29.img_norm1.weight",
+    "transformer_blocks.29.img_norm2.weight",
+    "transformer_blocks.29.txt_norm1.weight",
+    "transformer_blocks.29.txt_norm2.weight",
+    "transformer_blocks.30.attn.norm_q.weight",
+    "transformer_blocks.30.attn.norm_k.weight",
+    "transformer_blocks.30.attn.norm_added_q.weight",
+    "transformer_blocks.30.attn.norm_added_k.weight",
+    "transformer_blocks.30.img_norm1.weight",
+    "transformer_blocks.30.img_norm2.weight",
+    "transformer_blocks.30.txt_norm1.weight",
+    "transformer_blocks.30.txt_norm2.weight",
+    "transformer_blocks.31.attn.norm_q.weight",
+    "transformer_blocks.31.attn.norm_k.weight",
+    "transformer_blocks.31.attn.norm_added_q.weight",
+    "transformer_blocks.31.attn.norm_added_k.weight",
+    "transformer_blocks.31.img_norm1.weight",
+    "transformer_blocks.31.img_norm2.weight",
+    "transformer_blocks.31.txt_norm1.weight",
+    "transformer_blocks.31.txt_norm2.weight",
+    "transformer_blocks.32.attn.norm_q.weight",
+    "transformer_blocks.32.attn.norm_k.weight",
+    "transformer_blocks.32.attn.norm_added_q.weight",
+    "transformer_blocks.32.attn.norm_added_k.weight",
+    "transformer_blocks.32.img_norm1.weight",
+    "transformer_blocks.32.img_norm2.weight",
+    "transformer_blocks.32.txt_norm1.weight",
+    "transformer_blocks.32.txt_norm2.weight",
+    "transformer_blocks.33.attn.norm_q.weight",
+    "transformer_blocks.33.attn.norm_k.weight",
+    "transformer_blocks.33.attn.norm_added_q.weight",
+    "transformer_blocks.33.attn.norm_added_k.weight",
+    "transformer_blocks.33.img_norm1.weight",
+    "transformer_blocks.33.img_norm2.weight",
+    "transformer_blocks.33.txt_norm1.weight",
+    "transformer_blocks.33.txt_norm2.weight",
+    "transformer_blocks.34.attn.norm_q.weight",
+    "transformer_blocks.34.attn.norm_k.weight",
+    "transformer_blocks.34.attn.norm_added_q.weight",
+    "transformer_blocks.34.attn.norm_added_k.weight",
+    "transformer_blocks.34.img_norm1.weight",
+    "transformer_blocks.34.img_norm2.weight",
+    "transformer_blocks.34.txt_norm1.weight",
+    "transformer_blocks.34.txt_norm2.weight",
+    "transformer_blocks.35.attn.norm_q.weight",
+    "transformer_blocks.35.attn.norm_k.weight",
+    "transformer_blocks.35.attn.norm_added_q.weight",
+    "transformer_blocks.35.attn.norm_added_k.weight",
+    "transformer_blocks.35.img_norm1.weight",
+    "transformer_blocks.35.img_norm2.weight",
+    "transformer_blocks.35.txt_norm1.weight",
+    "transformer_blocks.35.txt_norm2.weight",
+    "transformer_blocks.36.attn.norm_q.weight",
+    "transformer_blocks.36.attn.norm_k.weight",
+    "transformer_blocks.36.attn.norm_added_q.weight",
+    "transformer_blocks.36.attn.norm_added_k.weight",
+    "transformer_blocks.36.img_norm1.weight",
+    "transformer_blocks.36.img_norm2.weight",
+    "transformer_blocks.36.txt_norm1.weight",
+    "transformer_blocks.36.txt_norm2.weight",
+    "transformer_blocks.37.attn.norm_q.weight",
+    "transformer_blocks.37.attn.norm_k.weight",
+    "transformer_blocks.37.attn.norm_added_q.weight",
+    "transformer_blocks.37.attn.norm_added_k.weight",
+    "transformer_blocks.37.img_norm1.weight",
+    "transformer_blocks.37.img_norm2.weight",
+    "transformer_blocks.37.txt_norm1.weight",
+    "transformer_blocks.37.txt_norm2.weight",
+    "transformer_blocks.38.attn.norm_q.weight",
+    "transformer_blocks.38.attn.norm_k.weight",
+    "transformer_blocks.38.attn.norm_added_q.weight",
+    "transformer_blocks.38.attn.norm_added_k.weight",
+    "transformer_blocks.38.img_norm1.weight",
+    "transformer_blocks.38.img_norm2.weight",
+    "transformer_blocks.38.txt_norm1.weight",
+    "transformer_blocks.38.txt_norm2.weight",
+    "transformer_blocks.39.attn.norm_q.weight",
+    "transformer_blocks.39.attn.norm_k.weight",
+    "transformer_blocks.39.attn.norm_added_q.weight",
+    "transformer_blocks.39.attn.norm_added_k.weight",
+    "transformer_blocks.39.img_norm1.weight",
+    "transformer_blocks.39.img_norm2.weight",
+    "transformer_blocks.39.txt_norm1.weight",
+    "transformer_blocks.39.txt_norm2.weight",
+    "transformer_blocks.40.attn.norm_q.weight",
+    "transformer_blocks.40.attn.norm_k.weight",
+    "transformer_blocks.40.attn.norm_added_q.weight",
+    "transformer_blocks.40.attn.norm_added_k.weight",
+    "transformer_blocks.40.img_norm1.weight",
+    "transformer_blocks.40.img_norm2.weight",
+    "transformer_blocks.40.txt_norm1.weight",
+    "transformer_blocks.40.txt_norm2.weight",
+    "transformer_blocks.41.attn.norm_q.weight",
+    "transformer_blocks.41.attn.norm_k.weight",
+    "transformer_blocks.41.attn.norm_added_q.weight",
+    "transformer_blocks.41.attn.norm_added_k.weight",
+    "transformer_blocks.41.img_norm1.weight",
+    "transformer_blocks.41.img_norm2.weight",
+    "transformer_blocks.41.txt_norm1.weight",
+    "transformer_blocks.41.txt_norm2.weight",
+    "transformer_blocks.42.attn.norm_q.weight",
+    "transformer_blocks.42.attn.norm_k.weight",
+    "transformer_blocks.42.attn.norm_added_q.weight",
+    "transformer_blocks.42.attn.norm_added_k.weight",
+    "transformer_blocks.42.img_norm1.weight",
+    "transformer_blocks.42.img_norm2.weight",
+    "transformer_blocks.42.txt_norm1.weight",
+    "transformer_blocks.42.txt_norm2.weight",
+    "transformer_blocks.43.attn.norm_q.weight",
+    "transformer_blocks.43.attn.norm_k.weight",
+    "transformer_blocks.43.attn.norm_added_q.weight",
+    "transformer_blocks.43.attn.norm_added_k.weight",
+    "transformer_blocks.43.img_norm1.weight",
+    "transformer_blocks.43.img_norm2.weight",
+    "transformer_blocks.43.txt_norm1.weight",
+    "transformer_blocks.43.txt_norm2.weight",
+    "transformer_blocks.44.attn.norm_q.weight",
+    "transformer_blocks.44.attn.norm_k.weight",
+    "transformer_blocks.44.attn.norm_added_q.weight",
+    "transformer_blocks.44.attn.norm_added_k.weight",
+    "transformer_blocks.44.img_norm1.weight",
+    "transformer_blocks.44.img_norm2.weight",
+    "transformer_blocks.44.txt_norm1.weight",
+    "transformer_blocks.44.txt_norm2.weight",
+    "transformer_blocks.45.attn.norm_q.weight",
+    "transformer_blocks.45.attn.norm_k.weight",
+    "transformer_blocks.45.attn.norm_added_q.weight",
+    "transformer_blocks.45.attn.norm_added_k.weight",
+    "transformer_blocks.45.img_norm1.weight",
+    "transformer_blocks.45.img_norm2.weight",
+    "transformer_blocks.45.txt_norm1.weight",
+    "transformer_blocks.45.txt_norm2.weight",
+    "transformer_blocks.46.attn.norm_q.weight",
+    "transformer_blocks.46.attn.norm_k.weight",
+    "transformer_blocks.46.attn.norm_added_q.weight",
+    "transformer_blocks.46.attn.norm_added_k.weight",
+    "transformer_blocks.46.img_norm1.weight",
+    "transformer_blocks.46.img_norm2.weight",
+    "transformer_blocks.46.txt_norm1.weight",
+    "transformer_blocks.46.txt_norm2.weight",
+    "transformer_blocks.47.attn.norm_q.weight",
+    "transformer_blocks.47.attn.norm_k.weight",
+    "transformer_blocks.47.attn.norm_added_q.weight",
+    "transformer_blocks.47.attn.norm_added_k.weight",
+    "transformer_blocks.47.img_norm1.weight",
+    "transformer_blocks.47.img_norm2.weight",
+    "transformer_blocks.47.txt_norm1.weight",
+    "transformer_blocks.47.txt_norm2.weight"
+  ],
+  "modules_to_not_use_matmul": [],
+  "non_blocking": false,
+  "quant_conv": false,
+  "quant_embedding": false,
+  "quant_method": "sdnq",
+  "quantization_device": null,
+  "quantized_matmul_dtype": "int8",
+  "return_device": null,
+  "sdnq_version": "0.2.0",
+  "svd_rank": 32,
+  "svd_steps": 8,
+  "use_dynamic_quantization": false,
+  "use_grad_ckpt": true,
+  "use_hadamard": false,
+  "use_quantized_matmul": true,
+  "use_quantized_matmul_conv": false,
+  "use_static_quantization": true,
+  "use_stochastic_rounding": false,
+  "use_svd": false,
+  "weights_dtype": "uint4"
+}

vae/config.json

@@ -0,0 +1,41 @@
+{
+  "_class_name": "AutoencoderKLFlux2",
+  "_diffusers_version": "0.38.0",
+  "_name_or_path": "/workspace/.cache/huggingface/hub/models--microsoft--Lens-Turbo/snapshots/f21b81bed3bae7f93f16f5fd300af9c408f5816a/vae",
+  "act_fn": "silu",
+  "batch_norm_eps": 0.0001,
+  "batch_norm_momentum": 0.1,
+  "block_out_channels": [
+    128,
+    256,
+    512,
+    512
+  ],
+  "decoder_block_out_channels": null,
+  "down_block_types": [
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D",
+    "DownEncoderBlock2D"
+  ],
+  "force_upcast": true,
+  "in_channels": 3,
+  "latent_channels": 32,
+  "layers_per_block": 2,
+  "mid_block_add_attention": true,
+  "norm_num_groups": 32,
+  "out_channels": 3,
+  "patch_size": [
+    2,
+    2
+  ],
+  "sample_size": 1024,
+  "up_block_types": [
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D",
+    "UpDecoderBlock2D"
+  ],
+  "use_post_quant_conv": true,
+  "use_quant_conv": true
+}

vae/diffusion_pytorch_model.safetensors

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