Hugging Face's logo

Baragi-AI
/
Anima-Style-Embedding

Feature Extraction
PyTorch
Safetensors
anima_style_embedding
anime
style-embedding
image-retrieval
reference-image
siglip2
anima
Model card Files
xet
 Community
Anima-Style-Embedding
File size: 1,372 Bytes
aa3d7f9
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
{
  "model_type": "anima_style_embedding",
  "version": "1.0",
  "architecture": {
    "width": 512,
    "blocks": 4,
    "heads": 8,
    "embedding_dim": 512,
    "dropout": 0.1,
    "reference_grounding": true,
    "functional_factors": true,
    "reference_count": [
      1,
      8
    ],
    "trainable_parameters": 17488353
  },
  "inputs": {
    "siglip2_full": {
      "shape": [
        30,
        1152
      ],
      "dtype": "bfloat16"
    },
    "siglip2_face": {
      "shape": [
        30,
        1152
      ],
      "dtype": "bfloat16",
      "optional": true
    },
    "anima": {
      "shape": [
        3,
        4096
      ],
      "dtype": "bfloat16"
    }
  },
  "vision_backbone": {
    "repo_id": "timm/vit_so400m_patch16_siglip_gap_512.v2_webli",
    "image_size": 512,
    "layer_fractions": [
      0.25,
      0.55,
      1.0
    ],
    "pool_sizes": [
      2,
      4,
      2
    ]
  },
  "anima_features": {
    "image_size": 768,
    "blocks": [
      8,
      18,
      26
    ],
    "sigma": 0.1,
    "noise_seed": 20260715
  },
  "training": {
    "checkpoint_step": 4500,
    "face_dropout": 0.3,
    "query_count": 2,
    "optimizer": "AdamW(fused)",
    "scheduler": "cosine_with_linear_warmup",
    "learning_rate": 0.0001,
    "packed_manifest_sha256": "c1267b9fd45c6e0d222b91d6ae2291b36b763277cc39cbc32484874ffeae9b32"
  }
}