SnJake commited on
Commit
bff283b
·
verified ·
1 Parent(s): 37a73e6

Add Baikal LoopSR v2 x2 EMA weights and updated examples

Browse files

Publish EMA update 30000 with its architecture config and paired examples; retain LoopSR v1 and SwinFIR v31.

Baikal_LoopSR_x2_v2.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a9ec49141e58b9ba109f3dde7d408dfccabbae08c0ba8fb3545ba3017e95f643
3
+ size 20954228
README.md CHANGED
@@ -1,45 +1,46 @@
1
- ---
2
- license: mit
3
- pipeline_tag: image-to-image
4
- library_name: pytorch
5
- tags: [super-resolution, image-restoration, anime, illustration, comfyui, transformer, loopsr, swinfir]
6
- ---
7
-
8
- # Baikal Anime Upscaler
9
-
10
- **x2 upscaling for anime and illustrations**, using Baikal LoopSR or SwinFIR v31.
11
-
12
- [ComfyUI nodes](https://github.com/SnJake/ComfyUI-Baikal-Anime/tree/dev) · [Example workflow](https://github.com/SnJake/ComfyUI-Baikal-Anime/blob/dev/workflows/Baikal_LoopSR_x2.json)
13
-
14
- | Model | Download |
15
- |---|---|
16
- | Baikal LoopSR x2 | [Baikal_LoopSR_x2.safetensors](Baikal_LoopSR_x2.safetensors) |
17
- | Baikal SwinFIR v31 | [Baikal_SwinFIR_Anime_x2_v31.safetensors](Baikal_SwinFIR_Anime_x2_v31.safetensors) |
18
-
19
- ## Examples
20
-
21
- **Bicubic x2 → Baikal LoopSR x2.** Same display size, no extra sharpening. LoopSR: EMA step 24000, loops 4.
22
-
23
- ![Illustration before and after](examples/illustration-overview.webp)
24
-
25
- ![Illustration details](examples/illustration-details.png)
26
-
27
- [Full input](examples/illustration-input.png) · [Full x2 result](examples/illustration-loopsr-x2.png)
28
-
29
- ![Anime before and after](examples/anime-overview.webp)
30
-
31
- ![Anime details](examples/anime-details.png)
32
-
33
- [Full input](examples/anime-input.png) · [Full x2 result](examples/anime-loopsr-x2.png)
34
-
35
- ## ComfyUI
36
-
37
- Install [ComfyUI-Baikal-Anime](https://github.com/SnJake/ComfyUI-Baikal-Anime/tree/dev), restart, and place weights in **`ComfyUI/models/anime_upscale/`**. The loader can also download them.
38
-
39
- Category **`Baikal/Upscale`**: **Baikal Model Loader → Baikal Anime Upscale → Save Image**.
40
-
41
- LoopSR supports **loops 1–4**. Default LR tile/overlap/halo: **256 / 32 / 96**. The custom loader also provides `UPSCALE_MODEL` for the standard upscaling node.
42
-
43
- LoopSR adapts shared loops, XSA and deep supervision from [Looped Diffusion Transformer](https://arxiv.org/abs/2609.40305). [Technical notes](https://github.com/SnJake/ComfyUI-Baikal-Anime/blob/dev/docs/TRAINING.md).
44
-
45
- MIT · [License](LICENSE.md)
 
 
1
+ ---
2
+ license: mit
3
+ pipeline_tag: image-to-image
4
+ library_name: pytorch
5
+ tags: [super-resolution, image-restoration, anime, illustration, comfyui, transformer, loopsr, swinfir]
6
+ ---
7
+
8
+ # Baikal Anime Upscaler
9
+
10
+ **x2 upscaling for anime and illustrations**, using Baikal LoopSR v2, LoopSR v1 or SwinFIR v31.
11
+
12
+ [ComfyUI nodes](https://github.com/SnJake/ComfyUI-Baikal-Anime/tree/dev) · [Example workflow](https://github.com/SnJake/ComfyUI-Baikal-Anime/blob/dev/workflows/Baikal_LoopSR_x2_v2.json)
13
+
14
+ | Model | Download |
15
+ |---|---|
16
+ | Baikal LoopSR v2 x2 | [Baikal_LoopSR_x2_v2.safetensors](https://huggingface.co/SnJake/Baikal-Anime-Upscaler/resolve/dev/Baikal_LoopSR_x2_v2.safetensors?download=true) |
17
+ | Baikal LoopSR v1 x2 | [Baikal_LoopSR_x2.safetensors](Baikal_LoopSR_x2.safetensors) |
18
+ | Baikal SwinFIR v31 | [Baikal_SwinFIR_Anime_x2_v31.safetensors](Baikal_SwinFIR_Anime_x2_v31.safetensors) |
19
+
20
+ ## Examples
21
+
22
+ **Bicubic x2 → Baikal LoopSR v2 x2.** Same display size, no extra sharpening. V2: EMA step 30000, loops 4.
23
+
24
+ ![Illustration before and after](examples/illustration-overview.webp)
25
+
26
+ ![Illustration details](examples/illustration-details.png)
27
+
28
+ [Full input](examples/illustration-input.png) · [Full x2 result](examples/illustration-loopsr-x2.png)
29
+
30
+ ![Anime before and after](examples/anime-overview.webp)
31
+
32
+ ![Anime details](examples/anime-details.png)
33
+
34
+ [Full input](examples/anime-input.png) · [Full x2 result](examples/anime-loopsr-x2.png)
35
+
36
+ ## ComfyUI
37
+
38
+ Install [ComfyUI-Baikal-Anime](https://github.com/SnJake/ComfyUI-Baikal-Anime/tree/dev), restart, and place weights in **`ComfyUI/models/anime_upscale/`**. The loader can also download them.
39
+
40
+ Category **`Baikal/Upscale`**: **Baikal Model Loader → Baikal Anime Upscale → Save Image**.
41
+
42
+ LoopSR supports **loops 1–4**. Default LR tile/overlap/halo: **256 / 32 / 96**. The custom loader also provides `UPSCALE_MODEL` for the standard upscaling node.
43
+
44
+ LoopSR adapts shared loops, XSA and deep supervision from [Looped Diffusion Transformer](https://arxiv.org/abs/2609.40305). [Technical notes](https://github.com/SnJake/ComfyUI-Baikal-Anime/blob/dev/docs/TRAINING.md).
45
+
46
+ MIT · [License](LICENSE.md)
examples/anime-details.png CHANGED

Git LFS Details

  • SHA256: 4d019db4cce504ac4dc4679897f4a2ff5a2b2b7d1a513750bff0af3be334cfc1
  • Pointer size: 131 Bytes
  • Size of remote file: 396 kB

Git LFS Details

  • SHA256: 3538806dab99e64d7946629f90a1a680b6f996a1c5e14b7d3c3ad8fe69eed3b3
  • Pointer size: 131 Bytes
  • Size of remote file: 396 kB
examples/anime-loopsr-x2.png CHANGED

Git LFS Details

  • SHA256: f3f0b88b22015b5885654c9fd0ccec4912b74120e9288ca1d36a89e8742d2eb8
  • Pointer size: 132 Bytes
  • Size of remote file: 3.58 MB

Git LFS Details

  • SHA256: 6ef1a2a92a4e9790c26c4c58a31a07b81aeceb5d2de3e4fe135d19d517936143
  • Pointer size: 132 Bytes
  • Size of remote file: 3.57 MB
examples/anime-overview.webp CHANGED

Git LFS Details

  • SHA256: c7985bffa233a1b619b410b8be7d744c75cec7f6719db362bd4da58789f451f0
  • Pointer size: 131 Bytes
  • Size of remote file: 551 kB

Git LFS Details

  • SHA256: 74bb0977c316ca8ccbb0d5427a5550d66f2f6b8cceffd59b4998371435198edf
  • Pointer size: 131 Bytes
  • Size of remote file: 548 kB
examples/illustration-details.png CHANGED

Git LFS Details

  • SHA256: 360d1a3a4a3993002f1787108d35cc616e0783c88d67b67b347bbd369818ecc0
  • Pointer size: 131 Bytes
  • Size of remote file: 370 kB

Git LFS Details

  • SHA256: a96657c25657953e58301322848336651cbf41b66c731ef0560cff3f26b898fc
  • Pointer size: 131 Bytes
  • Size of remote file: 377 kB
examples/illustration-loopsr-x2.png CHANGED

Git LFS Details

  • SHA256: fd6b8d3e6ce4bec3fc2268303ea6ce5765da07bf2c164a3be5e2a091f6767645
  • Pointer size: 132 Bytes
  • Size of remote file: 8.75 MB

Git LFS Details

  • SHA256: 9a6f67615a1e06f4937be67fd4954f1b8336a650dafe7cd84895202e1d3c9551
  • Pointer size: 132 Bytes
  • Size of remote file: 8.94 MB
examples/illustration-overview.webp CHANGED

Git LFS Details

  • SHA256: 72a879f1d00d4ef1d6874a8c0f2fe63186405d11954589a309c1d1dab326c196
  • Pointer size: 132 Bytes
  • Size of remote file: 1.04 MB

Git LFS Details

  • SHA256: 9c88306296c40ff5b1aab47d9fee65a10bf6a2a223fa9c779d3ead52e4e5f3b6
  • Pointer size: 132 Bytes
  • Size of remote file: 1.04 MB
examples/metadata.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
- "model": "Baikal_LoopSR_x2.safetensors",
3
  "weights": "EMA",
4
- "training_steps": 24000,
5
  "baseline": "bicubic x2, not an HR reference",
6
  "examples": [
7
  {
 
1
  {
2
+ "model": "Baikal_LoopSR_x2_v2.safetensors",
3
  "weights": "EMA",
4
+ "training_steps": 30000,
5
  "baseline": "bicubic x2, not an HR reference",
6
  "examples": [
7
  {
loopsr_x2_v2_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architecture": "Baikal_LoopSR",
3
+ "model_version": 2,
4
+ "model": {
5
+ "scale": 2,
6
+ "dim": 224,
7
+ "heads": 7,
8
+ "window": 8,
9
+ "pre_blocks": 3,
10
+ "loop_blocks": 4,
11
+ "post_blocks": 2,
12
+ "loops": 4,
13
+ "ffn_ratio": 2.0,
14
+ "attention": "xsa",
15
+ "checkpoint_blocks": false
16
+ },
17
+ "training_steps": 30000,
18
+ "weights": "EMA"
19
+ }