recoilme commited on
Commit
7222001
·
1 Parent(s): cf8040d

diffusers

Browse files
.gitattributes CHANGED
@@ -35,3 +35,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  *.jpg filter=lfs diff=lfs merge=lfs -text
37
  *.png filter=lfs diff=lfs merge=lfs -text
 
 
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
  *.jpg filter=lfs diff=lfs merge=lfs -text
37
  *.png filter=lfs diff=lfs merge=lfs -text
38
+ tokenizer/*.json filter=lfs diff=lfs merge=lfs -text
samples/unet_192x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 9c41a02bc019e960f68a1755a258da951838125eb4b9c93c63ca6ac7a70122db
  • Pointer size: 130 Bytes
  • Size of remote file: 49.8 kB

Git LFS Details

  • SHA256: e619c86e7b9ef4f0c1031d6d9afb584e020fa986147b8bcbb4c4d21935cd23be
  • Pointer size: 130 Bytes
  • Size of remote file: 41.7 kB
samples/unet_256x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 0dcb38fdf7b9a8510d025be1c31e990119ba14d0173690ce9c543198ce04d8d2
  • Pointer size: 130 Bytes
  • Size of remote file: 57 kB

Git LFS Details

  • SHA256: 6a444c56913d7b23d148d1417ed0b2e9d90256da09d955035adc942720e67ecb
  • Pointer size: 130 Bytes
  • Size of remote file: 55.9 kB
samples/unet_320x384_0.jpg CHANGED

Git LFS Details

  • SHA256: 5f6b893b2aaacd9c8586ec369baad9aa65a0e0e99488d5e3ff4d6216cda0d55c
  • Pointer size: 130 Bytes
  • Size of remote file: 68.1 kB

Git LFS Details

  • SHA256: 89c605a8ad57ce0ae05a10278f249598b749f3c15b01e92b9b90d99b0957b0bb
  • Pointer size: 130 Bytes
  • Size of remote file: 72.5 kB
samples/unet_384x192_0.jpg CHANGED

Git LFS Details

  • SHA256: 38dd60aaac6c66723104db6c319bbe4fc8d4625088641f34a0dcd9e1e80462e5
  • Pointer size: 130 Bytes
  • Size of remote file: 36.5 kB

Git LFS Details

  • SHA256: 7018e6608421282d7ca6b15941a3577e905db6684f8b410877f7b2c2ef8fa5ff
  • Pointer size: 130 Bytes
  • Size of remote file: 39.9 kB
samples/unet_384x256_0.jpg CHANGED

Git LFS Details

  • SHA256: eb31498923fa15edfb0bacb5690546692b424892f9482c92776c8ecf54c3f2eb
  • Pointer size: 130 Bytes
  • Size of remote file: 69.9 kB

Git LFS Details

  • SHA256: cb1d7a5789ea691346f1de0ca0ce13e296fb8d5217ed44855fb3635838376e25
  • Pointer size: 130 Bytes
  • Size of remote file: 69 kB
samples/unet_384x320_0.jpg CHANGED

Git LFS Details

  • SHA256: d8ff680aef19690d10d1efcd4698ffbade22190331070724ce78c110cfb7cd29
  • Pointer size: 130 Bytes
  • Size of remote file: 61.5 kB

Git LFS Details

  • SHA256: 8c8fbc912e72d244de372dbfc6638ee0fffd8cc0963aaa9aae5c7f71f7fbd01a
  • Pointer size: 130 Bytes
  • Size of remote file: 73.4 kB
samples/unet_384x384_0.jpg CHANGED

Git LFS Details

  • SHA256: e6e5fd88fdc1d4b4791993f83ca300a9a6716d9ff58929fdb1bc07cb9267a69f
  • Pointer size: 130 Bytes
  • Size of remote file: 70.6 kB

Git LFS Details

  • SHA256: 5fc1fdec90eab6b90b87dfd4d4ad3319b7321732bb021d887fdbecb9e89f58aa
  • Pointer size: 130 Bytes
  • Size of remote file: 76.4 kB
src/sample.ipynb CHANGED
@@ -276,9 +276,42 @@
276
  },
277
  {
278
  "cell_type": "code",
279
- "execution_count": null,
280
  "id": "f4614e4b-2abd-453a-9703-65697c1d56bb",
281
  "metadata": {},
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
282
  "outputs": [],
283
  "source": []
284
  }
 
276
  },
277
  {
278
  "cell_type": "code",
279
+ "execution_count": 1,
280
  "id": "f4614e4b-2abd-453a-9703-65697c1d56bb",
281
  "metadata": {},
282
+ "outputs": [
283
+ {
284
+ "name": "stdout",
285
+ "output_type": "stream",
286
+ "text": [
287
+ "ok\n"
288
+ ]
289
+ }
290
+ ],
291
+ "source": [
292
+ "import torch\n",
293
+ "from diffusers import AutoencoderKL, UNet2DConditionModel, FlowMatchEulerDiscreteScheduler\n",
294
+ "from transformers import AutoTokenizer, AutoModel\n",
295
+ "from PIL import Image\n",
296
+ "import math\n",
297
+ "\n",
298
+ "# ====== Настройки ======\n",
299
+ "dtype = torch.float16\n",
300
+ "device = torch.device(\"cuda\" if torch.cuda.is_available() else \"cpu\")\n",
301
+ "\n",
302
+ "unet = UNet2DConditionModel.from_pretrained( \"/workspace/sdxs3d\"#\"AiArtLab/sdxs3d\"\n",
303
+ " , subfolder=\"unet\", torch_dtype=dtype\n",
304
+ ").to(device).eval()\n",
305
+ "\n",
306
+ "unet.save_pretrained(\"/workspace/sdxs3d/unet\")\n",
307
+ "print(\"ok\")"
308
+ ]
309
+ },
310
+ {
311
+ "cell_type": "code",
312
+ "execution_count": null,
313
+ "id": "b08fbf66-8bd1-4a20-8715-0e748a07a932",
314
+ "metadata": {},
315
  "outputs": [],
316
  "source": []
317
  }
tokenizer/added_tokens.json CHANGED
@@ -1,28 +1,3 @@
1
- {
2
- "</think>": 151668,
3
- "</tool_call>": 151658,
4
- "</tool_response>": 151666,
5
- "<think>": 151667,
6
- "<tool_call>": 151657,
7
- "<tool_response>": 151665,
8
- "<|box_end|>": 151649,
9
- "<|box_start|>": 151648,
10
- "<|endoftext|>": 151643,
11
- "<|file_sep|>": 151664,
12
- "<|fim_middle|>": 151660,
13
- "<|fim_pad|>": 151662,
14
- "<|fim_prefix|>": 151659,
15
- "<|fim_suffix|>": 151661,
16
- "<|im_end|>": 151645,
17
- "<|im_start|>": 151644,
18
- "<|image_pad|>": 151655,
19
- "<|object_ref_end|>": 151647,
20
- "<|object_ref_start|>": 151646,
21
- "<|quad_end|>": 151651,
22
- "<|quad_start|>": 151650,
23
- "<|repo_name|>": 151663,
24
- "<|video_pad|>": 151656,
25
- "<|vision_end|>": 151653,
26
- "<|vision_pad|>": 151654,
27
- "<|vision_start|>": 151652
28
- }
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c0284b582e14987fbd3d5a2cb2bd139084371ed9acbae488829a1c900833c680
3
+ size 707
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
tokenizer/special_tokens_map.json CHANGED
@@ -1,31 +1,3 @@
1
- {
2
- "additional_special_tokens": [
3
- "<|im_start|>",
4
- "<|im_end|>",
5
- "<|object_ref_start|>",
6
- "<|object_ref_end|>",
7
- "<|box_start|>",
8
- "<|box_end|>",
9
- "<|quad_start|>",
10
- "<|quad_end|>",
11
- "<|vision_start|>",
12
- "<|vision_end|>",
13
- "<|vision_pad|>",
14
- "<|image_pad|>",
15
- "<|video_pad|>"
16
- ],
17
- "eos_token": {
18
- "content": "<|im_end|>",
19
- "lstrip": false,
20
- "normalized": false,
21
- "rstrip": false,
22
- "single_word": false
23
- },
24
- "pad_token": {
25
- "content": "<|endoftext|>",
26
- "lstrip": false,
27
- "normalized": false,
28
- "rstrip": false,
29
- "single_word": false
30
- }
31
- }
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76862e765266b85aa9459767e33cbaf13970f327a0e88d1c65846c2ddd3a1ecd
3
+ size 613
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
tokenizer/tokenizer_config.json CHANGED
@@ -1,240 +1,3 @@
1
- {
2
- "add_bos_token": false,
3
- "add_prefix_space": false,
4
- "added_tokens_decoder": {
5
- "151643": {
6
- "content": "<|endoftext|>",
7
- "lstrip": false,
8
- "normalized": false,
9
- "rstrip": false,
10
- "single_word": false,
11
- "special": true
12
- },
13
- "151644": {
14
- "content": "<|im_start|>",
15
- "lstrip": false,
16
- "normalized": false,
17
- "rstrip": false,
18
- "single_word": false,
19
- "special": true
20
- },
21
- "151645": {
22
- "content": "<|im_end|>",
23
- "lstrip": false,
24
- "normalized": false,
25
- "rstrip": false,
26
- "single_word": false,
27
- "special": true
28
- },
29
- "151646": {
30
- "content": "<|object_ref_start|>",
31
- "lstrip": false,
32
- "normalized": false,
33
- "rstrip": false,
34
- "single_word": false,
35
- "special": true
36
- },
37
- "151647": {
38
- "content": "<|object_ref_end|>",
39
- "lstrip": false,
40
- "normalized": false,
41
- "rstrip": false,
42
- "single_word": false,
43
- "special": true
44
- },
45
- "151648": {
46
- "content": "<|box_start|>",
47
- "lstrip": false,
48
- "normalized": false,
49
- "rstrip": false,
50
- "single_word": false,
51
- "special": true
52
- },
53
- "151649": {
54
- "content": "<|box_end|>",
55
- "lstrip": false,
56
- "normalized": false,
57
- "rstrip": false,
58
- "single_word": false,
59
- "special": true
60
- },
61
- "151650": {
62
- "content": "<|quad_start|>",
63
- "lstrip": false,
64
- "normalized": false,
65
- "rstrip": false,
66
- "single_word": false,
67
- "special": true
68
- },
69
- "151651": {
70
- "content": "<|quad_end|>",
71
- "lstrip": false,
72
- "normalized": false,
73
- "rstrip": false,
74
- "single_word": false,
75
- "special": true
76
- },
77
- "151652": {
78
- "content": "<|vision_start|>",
79
- "lstrip": false,
80
- "normalized": false,
81
- "rstrip": false,
82
- "single_word": false,
83
- "special": true
84
- },
85
- "151653": {
86
- "content": "<|vision_end|>",
87
- "lstrip": false,
88
- "normalized": false,
89
- "rstrip": false,
90
- "single_word": false,
91
- "special": true
92
- },
93
- "151654": {
94
- "content": "<|vision_pad|>",
95
- "lstrip": false,
96
- "normalized": false,
97
- "rstrip": false,
98
- "single_word": false,
99
- "special": true
100
- },
101
- "151655": {
102
- "content": "<|image_pad|>",
103
- "lstrip": false,
104
- "normalized": false,
105
- "rstrip": false,
106
- "single_word": false,
107
- "special": true
108
- },
109
- "151656": {
110
- "content": "<|video_pad|>",
111
- "lstrip": false,
112
- "normalized": false,
113
- "rstrip": false,
114
- "single_word": false,
115
- "special": true
116
- },
117
- "151657": {
118
- "content": "<tool_call>",
119
- "lstrip": false,
120
- "normalized": false,
121
- "rstrip": false,
122
- "single_word": false,
123
- "special": false
124
- },
125
- "151658": {
126
- "content": "</tool_call>",
127
- "lstrip": false,
128
- "normalized": false,
129
- "rstrip": false,
130
- "single_word": false,
131
- "special": false
132
- },
133
- "151659": {
134
- "content": "<|fim_prefix|>",
135
- "lstrip": false,
136
- "normalized": false,
137
- "rstrip": false,
138
- "single_word": false,
139
- "special": false
140
- },
141
- "151660": {
142
- "content": "<|fim_middle|>",
143
- "lstrip": false,
144
- "normalized": false,
145
- "rstrip": false,
146
- "single_word": false,
147
- "special": false
148
- },
149
- "151661": {
150
- "content": "<|fim_suffix|>",
151
- "lstrip": false,
152
- "normalized": false,
153
- "rstrip": false,
154
- "single_word": false,
155
- "special": false
156
- },
157
- "151662": {
158
- "content": "<|fim_pad|>",
159
- "lstrip": false,
160
- "normalized": false,
161
- "rstrip": false,
162
- "single_word": false,
163
- "special": false
164
- },
165
- "151663": {
166
- "content": "<|repo_name|>",
167
- "lstrip": false,
168
- "normalized": false,
169
- "rstrip": false,
170
- "single_word": false,
171
- "special": false
172
- },
173
- "151664": {
174
- "content": "<|file_sep|>",
175
- "lstrip": false,
176
- "normalized": false,
177
- "rstrip": false,
178
- "single_word": false,
179
- "special": false
180
- },
181
- "151665": {
182
- "content": "<tool_response>",
183
- "lstrip": false,
184
- "normalized": false,
185
- "rstrip": false,
186
- "single_word": false,
187
- "special": false
188
- },
189
- "151666": {
190
- "content": "</tool_response>",
191
- "lstrip": false,
192
- "normalized": false,
193
- "rstrip": false,
194
- "single_word": false,
195
- "special": false
196
- },
197
- "151667": {
198
- "content": "<think>",
199
- "lstrip": false,
200
- "normalized": false,
201
- "rstrip": false,
202
- "single_word": false,
203
- "special": false
204
- },
205
- "151668": {
206
- "content": "</think>",
207
- "lstrip": false,
208
- "normalized": false,
209
- "rstrip": false,
210
- "single_word": false,
211
- "special": false
212
- }
213
- },
214
- "additional_special_tokens": [
215
- "<|im_start|>",
216
- "<|im_end|>",
217
- "<|object_ref_start|>",
218
- "<|object_ref_end|>",
219
- "<|box_start|>",
220
- "<|box_end|>",
221
- "<|quad_start|>",
222
- "<|quad_end|>",
223
- "<|vision_start|>",
224
- "<|vision_end|>",
225
- "<|vision_pad|>",
226
- "<|image_pad|>",
227
- "<|video_pad|>"
228
- ],
229
- "bos_token": null,
230
- "clean_up_tokenization_spaces": false,
231
- "eos_token": "<|im_end|>",
232
- "errors": "replace",
233
- "extra_special_tokens": {},
234
- "model_max_length": 131072,
235
- "pad_token": "<|endoftext|>",
236
- "padding_side": "left",
237
- "split_special_tokens": false,
238
- "tokenizer_class": "Qwen2Tokenizer",
239
- "unk_token": null
240
- }
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ffe7b045351d007df591529681cad1b162859afaa83abc082da65c7e8eb994da
3
+ size 5430
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
tokenizer/vocab.json CHANGED
The diff for this file is too large to render. See raw diff
 
unet/config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.35.1",
4
- "_name_or_path": "/workspace/sdxs3d",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
 
1
  {
2
  "_class_name": "UNet2DConditionModel",
3
  "_diffusers_version": "0.35.1",
4
+ "_name_or_path": "unet",
5
  "act_fn": "silu",
6
  "addition_embed_type": null,
7
  "addition_embed_type_num_heads": 64,
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:7551366a018198ef740aa9128fa0b87a8adb41981a6945873fbf986d481c3377
3
- size 3092571208
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa7a39ef8639e7491e8673a47c66cbd2f8d772437619e65343cdf7620bb28bfc
3
+ size 6184944280