Flulike99 commited on
Commit
9b3cd57
·
1 Parent(s): 3fba8c8
.gitignore ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ _pycache__/
2
+ flux/_pycache__/
3
+ .python-version
4
+ .requirements.txt
5
+ run.sh
__pycache__/app.cpython-38.pyc ADDED
Binary file (11.1 kB). View file
 
app.py CHANGED
@@ -98,6 +98,33 @@ def _to_pil_rgba(img: Any) -> Image.Image:
98
  pil = pil.convert("RGBA")
99
  return pil
100
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
101
  def _place_subject_on_canvas(
102
  subject_rgba: Image.Image,
103
  canvas_size: int,
@@ -212,10 +239,14 @@ def apply_style(image: Image.Image, style: str, width: int = 1024, height: int =
212
 
213
  def generate_background_local(styled_image: Image.Image, prompt: str, steps: int = 10, width: int = 1024, height: int = 1024) -> Image.Image:
214
  """Generate background using local FLUX model"""
 
 
 
215
  if not FLUX_AVAILABLE:
216
  # Return a simple gradient background if FLUX is not available
217
  if styled_image is None:
218
  return Image.new("RGB", (width, height), (200, 200, 255))
 
219
  # Create a simple colored background
220
  bg = Image.new("RGB", (width, height), (200, 220, 255))
221
  if styled_image.mode == "RGBA":
@@ -228,6 +259,9 @@ def generate_background_local(styled_image: Image.Image, prompt: str, steps: int
228
 
229
  if styled_image is None:
230
  return Image.new("RGB", (width, height), (255, 255, 255))
 
 
 
231
 
232
  # Convert to RGB for background generation
233
  img_rgb = styled_image.convert("RGB")
@@ -387,6 +421,12 @@ def create_simple_app():
387
  # Ensure image is RGBA
388
  if image.mode != "RGBA":
389
  image = image.convert("RGBA")
 
 
 
 
 
 
390
 
391
  # Generate background using local model only
392
  return generate_background_local(image, prompt, steps, width, height)
 
98
  pil = pil.convert("RGBA")
99
  return pil
100
 
101
+ def _center_subject_on_canvas(subject_rgba: Image.Image, canvas_width: int, canvas_height: int) -> Image.Image:
102
+ """
103
+ Center the subject image on a transparent canvas that matches the requested size.
104
+ """
105
+ if subject_rgba is None:
106
+ return Image.new("RGBA", (canvas_width, canvas_height), (0, 0, 0, 0))
107
+
108
+ canvas = Image.new("RGBA", (canvas_width, canvas_height), (0, 0, 0, 0))
109
+ paste_x = (canvas_width - subject_rgba.width) // 2
110
+ paste_y = (canvas_height - subject_rgba.height) // 2
111
+
112
+ # If the subject is larger than canvas, crop to fit
113
+ if subject_rgba.width > canvas_width or subject_rgba.height > canvas_height:
114
+ subject_rgba = subject_rgba.crop(
115
+ (
116
+ max(0, -paste_x),
117
+ max(0, -paste_y),
118
+ max(0, -paste_x) + min(canvas_width, subject_rgba.width),
119
+ max(0, -paste_y) + min(canvas_height, subject_rgba.height),
120
+ )
121
+ )
122
+ paste_x = max(0, paste_x)
123
+ paste_y = max(0, paste_y)
124
+
125
+ canvas.alpha_composite(subject_rgba, dest=(paste_x, paste_y))
126
+ return canvas
127
+
128
  def _place_subject_on_canvas(
129
  subject_rgba: Image.Image,
130
  canvas_size: int,
 
239
 
240
  def generate_background_local(styled_image: Image.Image, prompt: str, steps: int = 10, width: int = 1024, height: int = 1024) -> Image.Image:
241
  """Generate background using local FLUX model"""
242
+ width = int(width)
243
+ height = int(height)
244
+
245
  if not FLUX_AVAILABLE:
246
  # Return a simple gradient background if FLUX is not available
247
  if styled_image is None:
248
  return Image.new("RGB", (width, height), (200, 200, 255))
249
+ styled_image = _center_subject_on_canvas(styled_image, width, height)
250
  # Create a simple colored background
251
  bg = Image.new("RGB", (width, height), (200, 220, 255))
252
  if styled_image.mode == "RGBA":
 
259
 
260
  if styled_image is None:
261
  return Image.new("RGB", (width, height), (255, 255, 255))
262
+
263
+ # Ensure the subject image matches requested canvas size
264
+ styled_image = _center_subject_on_canvas(styled_image, width, height)
265
 
266
  # Convert to RGB for background generation
267
  img_rgb = styled_image.convert("RGB")
 
421
  # Ensure image is RGBA
422
  if image.mode != "RGBA":
423
  image = image.convert("RGBA")
424
+
425
+ width = int(width)
426
+ height = int(height)
427
+
428
+ # Center original uploaded image on a transparent canvas without resizing
429
+ image = _center_subject_on_canvas(image, width, height)
430
 
431
  # Generate background using local model only
432
  return generate_background_local(image, prompt, steps, width, height)
flux/__pycache__/__init__.cpython-310.pyc ADDED
Binary file (126 Bytes). View file
 
flux/__pycache__/block.cpython-310.pyc ADDED
Binary file (5.93 kB). View file
 
flux/__pycache__/condition.cpython-310.pyc ADDED
Binary file (3.43 kB). View file
 
flux/__pycache__/generate.cpython-310.pyc ADDED
Binary file (7.57 kB). View file
 
flux/__pycache__/lora_controller.cpython-310.pyc ADDED
Binary file (3.09 kB). View file
 
flux/__pycache__/padding_orthogonalization.cpython-310.pyc ADDED
Binary file (6.55 kB). View file
 
flux/__pycache__/pipeline_tools.cpython-310.pyc ADDED
Binary file (2.11 kB). View file
 
flux/__pycache__/transformer.cpython-310.pyc ADDED
Binary file (4.23 kB). View file
 
requirements.txt CHANGED
@@ -1,943 +1,14 @@
1
- # This file was autogenerated by uv via the following command:
2
- # uv pip compile pyproject.toml
3
- absl-py==2.3.1
4
- # via
5
- # array-record
6
- # dm-tree
7
- # etils
8
- # keras
9
- # tensorboard
10
- # tensorflow
11
- # tensorflow-datasets
12
- # tensorflow-metadata
13
- accelerate==1.10.1
14
- # via peft
15
- aiofiles==24.1.0
16
- # via gradio
17
- aiohappyeyeballs==2.6.1
18
- # via aiohttp
19
- aiohttp==3.13.0
20
- # via
21
- # datasets
22
- # fsspec
23
- # gcsfs
24
- aiosignal==1.4.0
25
- # via aiohttp
26
- annotated-types==0.7.0
27
- # via pydantic
28
- antlr4-python3-runtime==4.9.3
29
- # via
30
- # hydra-core
31
- # omegaconf
32
- anyio==4.11.0
33
- # via
34
- # gradio
35
- # httpx
36
- # jupyter-server
37
- # starlette
38
- argon2-cffi==25.1.0
39
- # via jupyter-server
40
- argon2-cffi-bindings==25.1.0
41
- # via argon2-cffi
42
- array-record==0.8.1
43
- # via tensorflow-datasets
44
- arrow==1.3.0
45
- # via isoduration
46
- asttokens==3.0.0
47
- # via stack-data
48
- astunparse==1.6.3
49
- # via tensorflow
50
- async-lru==2.0.5
51
- # via jupyterlab
52
- attrs==25.4.0
53
- # via
54
- # aiohttp
55
- # dm-tree
56
- # jsonschema
57
- # referencing
58
- babel==2.17.0
59
- # via jupyterlab-server
60
- beautifulsoup4==4.14.2
61
- # via
62
- # gdown
63
- # nbconvert
64
- bitsandbytes==0.45.2
65
- # via ads-gen (pyproject.toml)
66
- bleach==6.2.0
67
- # via nbconvert
68
- brotli==1.1.0
69
- # via gradio
70
- cachetools==6.2.0
71
- # via google-auth
72
- certifi==2025.10.5
73
- # via
74
- # httpcore
75
- # httpx
76
- # requests
77
- cffi==2.0.0
78
- # via argon2-cffi-bindings
79
- charset-normalizer==3.4.3
80
- # via requests
81
- click==8.3.0
82
- # via
83
- # typer
84
- # uvicorn
85
- comm==0.2.3
86
- # via
87
- # ipykernel
88
- # ipywidgets
89
- # via matplotlib
90
- cycler==0.12.1
91
- # via matplotlib
92
- datasets==2.21.0
93
- # via ads-gen (pyproject.toml)
94
- debugpy==1.8.17
95
- # via ipykernel
96
- decorator==5.2.1
97
- # via
98
- # gcsfs
99
- # ipython
100
- defusedxml==0.7.1
101
- # via nbconvert
102
- diffusers==0.35.0
103
- # via ads-gen (pyproject.toml)
104
- dill==0.3.8
105
- # via
106
- # datasets
107
- # multiprocess
108
- dm-tree==0.1.9
109
- # via tensorflow-datasets
110
- docstring-parser==0.17.0
111
- # via simple-parsing
112
- dreamsim==0.2.1
113
- # via ads-gen (pyproject.toml)
114
- einops==0.6.1
115
- # via
116
- # ads-gen (pyproject.toml)
117
- # etils
118
- etils==1.13.0
119
- # via
120
- # array-record
121
- # tensorflow-datasets
122
- executing==2.2.1
123
- # via stack-data
124
- faiss-cpu==1.12.0
125
- # via ads-gen (pyproject.toml)
126
- fastapi==0.118.2
127
- # via gradio
128
- fastjsonschema==2.21.2
129
- # via nbformat
130
- ffmpy==0.6.2
131
- # via gradio
132
- filelock==3.20.0
133
- # via
134
- # datasets
135
- # diffusers
136
- # gdown
137
- # huggingface-hub
138
- # torch
139
- # transformers
140
- flatbuffers==25.9.23
141
- # via tensorflow
142
- fonttools==4.60.1
143
- # via matplotlib
144
- fqdn==1.5.1
145
- # via jsonschema
146
- frozenlist==1.8.0
147
- # via
148
- # aiohttp
149
- # aiosignal
150
- fsspec==2023.12.2
151
- # via
152
- # datasets
153
- # etils
154
- # gcsfs
155
- # gradio-client
156
- # huggingface-hub
157
- # torch
158
- ftfy==6.3.1
159
- # via open-clip-torch
160
- gast==0.6.0
161
- # via tensorflow
162
- gcsfs==2023.12.2.post1
163
- # via ads-gen (pyproject.toml)
164
- gdown==4.7.3
165
- # via ads-gen (pyproject.toml)
166
- google-api-core
167
- # via
168
- # google-cloud-core
169
- # google-cloud-storage
170
- google-auth==2.41.1
171
- # via
172
- # gcsfs
173
- # google-api-core
174
- # google-auth-oauthlib
175
- # google-cloud-core
176
- # google-cloud-storage
177
- google-auth-oauthlib==1.2.2
178
- # via gcsfs
179
- google-cloud-core==2.4.3
180
- # via google-cloud-storage
181
- google-cloud-storage==3.4.1
182
- # via gcsfs
183
- google-crc32c==1.7.1
184
- # via
185
- # google-cloud-storage
186
- # google-resumable-media
187
- google-pasta==0.2.0
188
- # via tensorflow
189
- google-resumable-media==2.7.2
190
- # via google-cloud-storage
191
- googleapis-common-protos
192
- # via
193
- # google-api-core
194
- # tensorflow-metadata
195
- gradio==5.49.1
196
- # via ads-gen (pyproject.toml)
197
- gradio-client==1.13.3
198
- # via gradio
199
- groovy==0.1.2
200
- # via gradio
201
- grpcio==1.75.1
202
- # via
203
- # tensorboard
204
- # tensorflow
205
- h11==0.16.0
206
- # via
207
- # httpcore
208
- # uvicorn
209
- h5py==3.14.0
210
- # via
211
- # keras
212
- # tensorflow
213
- hf-xet==1.1.10
214
- # via huggingface-hub
215
- httpcore==1.0.9
216
- # via httpx
217
- httpx==0.28.1
218
- # via
219
- # gradio
220
- # gradio-client
221
- # jupyterlab
222
- # safehttpx
223
- huggingface-hub==0.35.3
224
- # via
225
- # accelerate
226
- # datasets
227
- # diffusers
228
- # gradio
229
- # gradio-client
230
- # open-clip-torch
231
- # peft
232
- # timm
233
- # tokenizers
234
- # transformers
235
- hydra-core==1.3.2
236
- # via ads-gen (pyproject.toml)
237
- idna==3.10
238
- # via
239
- # anyio
240
- # httpx
241
- # jsonschema
242
- # requests
243
- # yarl
244
- imageio==2.37.0
245
- # via scikit-image
246
- immutabledict==4.2.1
247
- # via tensorflow-datasets
248
- importlib-metadata==8.7.0
249
- # via diffusers
250
- importlib-resources==6.5.2
251
- # via etils
252
- ipykernel==6.30.1
253
- # via
254
- # jupyter
255
- # jupyter-console
256
- # jupyterlab
257
- ipython
258
- # via
259
- # ipykernel
260
- # ipywidgets
261
- # jupyter-console
262
- ipython-pygments-lexers==1.1.1
263
- # via ipython
264
- ipywidgets==8.1.7
265
- # via jupyter
266
- isoduration==20.11.0
267
- # via jsonschema
268
- jedi==0.19.2
269
- # via ipython
270
- jinja2==3.1.6
271
- # via
272
- # gradio
273
- # jupyter-server
274
- # jupyterlab
275
- # jupyterlab-server
276
- # nbconvert
277
- # torch
278
- joblib==1.5.2
279
- # via
280
- # prdc
281
- # scikit-learn
282
- json5==0.12.1
283
- # via jupyterlab-server
284
- jsonpointer==3.0.0
285
- # via jsonschema
286
- jsonschema==4.25.1
287
- # via
288
- # jupyter-events
289
- # jupyterlab-server
290
- # nbformat
291
- jsonschema-specifications==2025.9.1
292
- # via jsonschema
293
- jupyter==1.1.1
294
- # via ads-gen (pyproject.toml)
295
- jupyter-client==8.6.3
296
- # via
297
- # ipykernel
298
- # jupyter-console
299
- # jupyter-server
300
- # nbclient
301
- jupyter-console==6.6.3
302
- # via jupyter
303
- jupyter-core==5.8.1
304
- # via
305
- # ipykernel
306
- # jupyter-client
307
- # jupyter-console
308
- # jupyter-server
309
- # jupyterlab
310
- # nbclient
311
- # nbconvert
312
- # nbformat
313
- jupyter-events==0.12.0
314
- # via jupyter-server
315
- jupyter-lsp==2.3.0
316
- # via jupyterlab
317
- jupyter-server==2.17.0
318
- # via
319
- # jupyter-lsp
320
- # jupyterlab
321
- # jupyterlab-server
322
- # notebook
323
- # notebook-shim
324
- jupyter-server-terminals==0.5.3
325
- # via jupyter-server
326
- jupyterlab==4.4.9
327
- # via
328
- # jupyter
329
- # notebook
330
- jupyterlab-pygments==0.3.0
331
- # via nbconvert
332
- jupyterlab-server==2.27.3
333
- # via
334
- # jupyterlab
335
- # notebook
336
- jupyterlab-widgets==3.0.15
337
- # via ipywidgets
338
- keras==3.11.3
339
- # via tensorflow
340
- kiwisolver==1.4.9
341
- # via matplotlib
342
- lark==1.3.0
343
- # via rfc3987-syntax
344
- lazy-loader==0.4
345
- # via scikit-image
346
- libclang==18.1.1
347
- # via tensorflow
348
- markdown==3.9
349
- # via tensorboard
350
- markdown-it-py==4.0.0
351
- # via rich
352
- markupsafe==3.0.3
353
- # via
354
- # gradio
355
- # jinja2
356
- # nbconvert
357
- # werkzeug
358
- matplotlib==3.10.7
359
- # via
360
- # ads-gen (pyproject.toml)
361
- # seaborn
362
- matplotlib-inline==0.1.7
363
- # via
364
- # ipykernel
365
- # ipython
366
- mdurl==0.1.2
367
- # via markdown-it-py
368
- mistune==3.1.4
369
- # via nbconvert
370
- ml-dtypes==0.5.3
371
- # via
372
- # keras
373
- # tensorflow
374
- mpmath==1.3.0
375
- # via sympy
376
- multidict==6.7.0
377
- # via
378
- # aiohttp
379
- # yarl
380
- multiprocess==0.70.16
381
- # via
382
- # ads-gen (pyproject.toml)
383
- # datasets
384
- namex==0.1.0
385
- # via keras
386
- nbclient==0.10.2
387
- # via nbconvert
388
- nbconvert==7.16.6
389
- # via
390
- # jupyter
391
- # jupyter-server
392
- nbformat==5.10.4
393
- # via
394
- # jupyter-server
395
- # nbclient
396
- # nbconvert
397
- nest-asyncio==1.6.0
398
- # via ipykernel
399
- networkx
400
- # via
401
- # scikit-image
402
- # torch
403
- notebook==7.4.7
404
- # via jupyter
405
- notebook-shim==0.2.4
406
- # via
407
- # jupyterlab
408
- # notebook
409
- numpy==1.26.4
410
- # via
411
- # accelerate
412
- # bitsandbytes
413
- # contourpy
414
- # datasets
415
- # diffusers
416
- # dm-tree
417
- # dreamsim
418
- # etils
419
- # faiss-cpu
420
- # gradio
421
- # h5py
422
- # imageio
423
- # keras
424
- # matplotlib
425
- # ml-dtypes
426
- # opencv-python
427
- # pandas
428
- # peft
429
- # prdc
430
- # pytorch-fid
431
- # scikit-image
432
- # scikit-learn
433
- # scipy
434
- # seaborn
435
- # tensorboard
436
- # tensorflow
437
- # tensorflow-datasets
438
- # tifffile
439
- # torchvision
440
- # transformers
441
- nvidia-cublas-cu12==12.6.4.1
442
- # via
443
- # nvidia-cudnn-cu12
444
- # nvidia-cusolver-cu12
445
- # torch
446
- nvidia-cuda-cupti-cu12==12.6.80
447
- # via torch
448
- nvidia-cuda-nvrtc-cu12==12.6.77
449
- # via torch
450
- nvidia-cuda-runtime-cu12==12.6.77
451
- # via torch
452
- nvidia-cudnn-cu12==9.5.1.17
453
- # via torch
454
- nvidia-cufft-cu12==11.3.0.4
455
- # via torch
456
- nvidia-cufile-cu12==1.11.1.6
457
- # via torch
458
- nvidia-curand-cu12==10.3.7.77
459
- # via torch
460
- nvidia-cusolver-cu12==11.7.1.2
461
- # via torch
462
- nvidia-cusparse-cu12==12.5.4.2
463
- # via
464
- # nvidia-cusolver-cu12
465
- # torch
466
- nvidia-cusparselt-cu12==0.6.3
467
- # via torch
468
- nvidia-ml-py==13.580.82
469
- # via nvitop
470
- nvidia-nccl-cu12==2.26.2
471
- # via torch
472
- nvidia-nvjitlink-cu12==12.6.85
473
- # via
474
- # nvidia-cufft-cu12
475
- # nvidia-cusolver-cu12
476
- # nvidia-cusparse-cu12
477
- # torch
478
- nvidia-nvtx-cu12==12.6.77
479
- # via torch
480
- nvitop==1.5.3
481
- # via ads-gen (pyproject.toml)
482
- oauthlib==3.3.1
483
- # via requests-oauthlib
484
- omegaconf==2.3.0
485
- # via
486
- # ads-gen (pyproject.toml)
487
- # hydra-core
488
- open-clip-torch==2.32.0
489
- # via dreamsim
490
- opencv-python==4.11.0.86
491
- # via ads-gen (pyproject.toml)
492
- opt-einsum==3.4.0
493
- # via tensorflow
494
- optree==0.17.0
495
- # via keras
496
- orjson==3.11.3
497
- # via gradio
498
- packaging==25.0
499
- # via
500
- # accelerate
501
- # datasets
502
- # faiss-cpu
503
- # gradio
504
- # gradio-client
505
- # huggingface-hub
506
- # hydra-core
507
- # ipykernel
508
- # jupyter-events
509
- # jupyter-server
510
- # jupyterlab
511
- # jupyterlab-server
512
- # keras
513
- # lazy-loader
514
- # matplotlib
515
- # nbconvert
516
- # peft
517
- # scikit-image
518
- # tensorboard
519
- # tensorflow
520
- # transformers
521
- pandas==2.3.3
522
- # via
523
- # datasets
524
- # gradio
525
- # seaborn
526
- # torch-tb-profiler
527
- pandocfilters==1.5.1
528
- # via nbconvert
529
- parso==0.8.5
530
- # via jedi
531
- peft==0.17.1
532
- # via
533
- # ads-gen (pyproject.toml)
534
- # dreamsim
535
- pexpect==4.9.0
536
- # via ipython
537
- pillow==9.5.0
538
- # via
539
- # ads-gen (pyproject.toml)
540
- # diffusers
541
- # dreamsim
542
- # gradio
543
- # imageio
544
- # matplotlib
545
- # pytorch-fid
546
- # scikit-image
547
- # tensorboard
548
- # torchvision
549
- platformdirs==4.5.0
550
- # via jupyter-core
551
- prdc==0.2
552
- # via ads-gen (pyproject.toml)
553
- prometheus-client==0.23.1
554
- # via jupyter-server
555
- promise==2.3
556
- # via tensorflow-datasets
557
- prompt-toolkit==3.0.52
558
- # via
559
- # ipython
560
- # jupyter-console
561
- propcache==0.4.1
562
- # via
563
- # aiohttp
564
- # yarl
565
- proto-plus
566
- # via google-api-core
567
- protobuf
568
- # via
569
- # ads-gen (pyproject.toml)
570
- # google-api-core
571
- # googleapis-common-protos
572
- # proto-plus
573
- # tensorboard
574
- # tensorflow
575
- # tensorflow-datasets
576
- # tensorflow-metadata
577
- psutil==7.1.0
578
- # via
579
- # accelerate
580
- # ipykernel
581
- # nvitop
582
- # peft
583
- # tensorflow-datasets
584
- ptyprocess==0.7.0
585
- # via
586
- # pexpect
587
- # terminado
588
- pure-eval==0.2.3
589
- # via stack-data
590
- pyarrow==21.0.0
591
- # via
592
- # datasets
593
- # tensorflow-datasets
594
- pyasn1==0.6.1
595
- # via
596
- # pyasn1-modules
597
- # rsa
598
- pyasn1-modules==0.4.2
599
- # via google-auth
600
- pycparser==2.23
601
- # via cffi
602
- pydantic==2.11.10
603
- # via
604
- # fastapi
605
- # gradio
606
- pydantic-core==2.33.2
607
- # via pydantic
608
- pydub==0.25.1
609
- # via gradio
610
- pygments==2.19.2
611
- # via
612
- # ipython
613
- # ipython-pygments-lexers
614
- # jupyter-console
615
- # nbconvert
616
- # rich
617
- pyparsing==3.2.5
618
- # via matplotlib
619
- pysocks==1.7.1
620
- # via requests
621
- python-dateutil==2.9.0.post0
622
- # via
623
- # arrow
624
- # jupyter-client
625
- # matplotlib
626
- # pandas
627
- python-json-logger==2.0.7
628
- # via
629
- # ads-gen (pyproject.toml)
630
- # jupyter-events
631
- python-multipart==0.0.20
632
- # via gradio
633
- pytorch-fid==0.3.0
634
- # via ads-gen (pyproject.toml)
635
- pytz==2025.2
636
- # via pandas
637
- pyyaml==6.0.3
638
- # via
639
- # ads-gen (pyproject.toml)
640
- # accelerate
641
- # datasets
642
- # gradio
643
- # huggingface-hub
644
- # jupyter-events
645
- # omegaconf
646
- # peft
647
- # timm
648
- # transformers
649
- pyzmq==27.1.0
650
- # via
651
- # ipykernel
652
- # jupyter-client
653
- # jupyter-console
654
- # jupyter-server
655
- referencing==0.36.2
656
- # via
657
- # jsonschema
658
- # jsonschema-specifications
659
- # jupyter-events
660
- regex==2025.9.18
661
- # via
662
- # diffusers
663
- # open-clip-torch
664
- # transformers
665
- requests==2.32.5
666
- # via
667
- # datasets
668
- # diffusers
669
- # fsspec
670
- # gcsfs
671
- # gdown
672
- # google-api-core
673
- # google-cloud-storage
674
- # huggingface-hub
675
- # jupyterlab-server
676
- # requests-oauthlib
677
- # tensorflow
678
- # tensorflow-datasets
679
- # transformers
680
- requests-oauthlib==2.0.0
681
- # via google-auth-oauthlib
682
- rfc3339-validator==0.1.4
683
- # via
684
- # jsonschema
685
- # jupyter-events
686
- rfc3986-validator==0.1.1
687
- # via
688
- # jsonschema
689
- # jupyter-events
690
- rfc3987-syntax==1.1.0
691
- # via jsonschema
692
- rich==13.9.4
693
- # via
694
- # ads-gen (pyproject.toml)
695
- # keras
696
- # typer
697
- rpds-py==0.27.1
698
- # via
699
- # jsonschema
700
- # referencing
701
- rsa==4.9.1
702
- # via google-auth
703
- ruff==0.14.0
704
- # via gradio
705
- safehttpx==0.1.6
706
- # via gradio
707
- safetensors==0.6.2
708
- # via
709
- # accelerate
710
- # diffusers
711
- # open-clip-torch
712
- # peft
713
- # timm
714
- # transformers
715
- scikit-image==0.24.0
716
- # via ads-gen (pyproject.toml)
717
- scikit-learn==1.7.2
718
- # via prdc
719
- scipy
720
- # via
721
- # ads-gen (pyproject.toml)
722
- # dreamsim
723
- # prdc
724
- # pytorch-fid
725
- # scikit-image
726
- # scikit-learn
727
- seaborn==0.12.2
728
- # via ads-gen (pyproject.toml)
729
- segment-anything @ git+https://github.com/facebookresearch/segment-anything.git@dca509fe793f601edb92606367a655c15ac00fdf
730
- # via ads-gen (pyproject.toml)
731
- semantic-version==2.10.0
732
- # via gradio
733
- send2trash==1.8.3
734
- # via jupyter-server
735
- sentencepiece==0.2.1
736
- # via ads-gen (pyproject.toml)
737
- setuptools==68.2.2
738
- # via
739
- # ads-gen (pyproject.toml)
740
- # jupyterlab
741
- # tensorboard
742
- # tensorflow
743
- # torch
744
- # triton
745
- shellingham==1.5.4
746
- # via typer
747
- simple-parsing==0.1.7
748
- # via tensorflow-datasets
749
- six==1.17.0
750
- # via
751
- # astunparse
752
- # gdown
753
- # google-pasta
754
- # promise
755
- # python-dateutil
756
- # rfc3339-validator
757
- # tensorflow
758
- sniffio==1.3.1
759
- # via anyio
760
- soupsieve==2.8
761
- # via beautifulsoup4
762
- stack-data==0.6.3
763
- # via ipython
764
- starlette==0.48.0
765
- # via
766
- # fastapi
767
- # gradio
768
- sympy==1.14.0
769
- # via torch
770
- tensorboard
771
- # via
772
- # ads-gen (pyproject.toml)
773
- # tensorflow
774
- # torch-tb-profiler
775
- tensorboard-data-server==0.7.2
776
- # via tensorboard
777
- tensorflow
778
- # via ads-gen (pyproject.toml)
779
- tensorflow-datasets
780
- # via ads-gen (pyproject.toml)
781
- tensorflow-metadata
782
- # via tensorflow-datasets
783
- termcolor==3.1.0
784
- # via
785
- # tensorflow
786
- # tensorflow-datasets
787
- terminado==0.18.1
788
- # via
789
- # jupyter-server
790
- # jupyter-server-terminals
791
- threadpoolctl==3.6.0
792
- # via scikit-learn
793
- tifffile
794
- # via scikit-image
795
- timm==0.9.16
796
- # via
797
- # ads-gen (pyproject.toml)
798
- # dreamsim
799
- # open-clip-torch
800
- tinycss2==1.4.0
801
- # via bleach
802
- tokenizers==0.22.1
803
- # via transformers
804
- toml==0.10.2
805
- # via tensorflow-datasets
806
- tomlkit==0.13.3
807
- # via gradio
808
  torch==2.7.0
809
- # via
810
- # accelerate
811
- # bitsandbytes
812
- # dreamsim
813
- # open-clip-torch
814
- # peft
815
- # pytorch-fid
816
- # timm
817
- # torchvision
818
- torch-tb-profiler==0.4.3
819
- # via ads-gen (pyproject.toml)
820
- torchao==0.13.0
821
- # via ads-gen (pyproject.toml)
822
- torchvision==0.22.0
823
- # via
824
- # ads-gen (pyproject.toml)
825
- # dreamsim
826
- # open-clip-torch
827
- # pytorch-fid
828
- # timm
829
- tornado==6.5.2
830
- # via
831
- # ipykernel
832
- # jupyter-client
833
- # jupyter-server
834
- # jupyterlab
835
- # notebook
836
- # terminado
837
- tqdm==4.67.1
838
- # via
839
- # datasets
840
- # etils
841
- # gdown
842
- # huggingface-hub
843
- # open-clip-torch
844
- # peft
845
- # tensorflow-datasets
846
- # transformers
847
- traitlets==5.14.3
848
- # via
849
- # ipykernel
850
- # ipython
851
- # ipywidgets
852
- # jupyter-client
853
- # jupyter-console
854
- # jupyter-core
855
- # jupyter-events
856
- # jupyter-server
857
- # jupyterlab
858
- # matplotlib-inline
859
- # nbclient
860
- # nbconvert
861
- # nbformat
862
- transformers==4.57.0
863
- # via
864
- # ads-gen (pyproject.toml)
865
- # dreamsim
866
- # peft
867
- triton==3.3.0
868
- # via
869
- # ads-gen (pyproject.toml)
870
- # torch
871
- typer==0.19.2
872
- # via gradio
873
- types-python-dateutil==2.9.0.20251008
874
- # via arrow
875
- typing-extensions==4.15.0
876
- # via
877
- # aiosignal
878
- # anyio
879
- # beautifulsoup4
880
- # etils
881
- # fastapi
882
- # gradio
883
- # gradio-client
884
- # grpcio
885
- # huggingface-hub
886
- # optree
887
- # pydantic
888
- # pydantic-core
889
- # referencing
890
- # simple-parsing
891
- # starlette
892
- # tensorflow
893
- # torch
894
- # typer
895
- # typing-inspection
896
- typing-inspection==0.4.2
897
- # via pydantic
898
- tzdata==2025.2
899
- # via pandas
900
- uri-template==1.3.0
901
- # via jsonschema
902
- urllib3==2.5.0
903
- # via requests
904
- uvicorn==0.37.0
905
- # via gradio
906
- wcwidth==0.2.14
907
- # via
908
- # ftfy
909
- # prompt-toolkit
910
- webcolors==24.11.1
911
- # via jsonschema
912
- webencodings==0.5.1
913
- # via
914
- # bleach
915
- # tinycss2
916
- websocket-client==1.9.0
917
- # via jupyter-server
918
- websockets==15.0.1
919
- # via gradio-client
920
- werkzeug==3.1.3
921
- # via tensorboard
922
- wheel==0.45.1
923
- # via astunparse
924
- widgetsnbextension==4.0.14
925
- # via ipywidgets
926
- wrapt==1.17.3
927
- # via
928
- # dm-tree
929
- # tensorflow
930
- # tensorflow-datasets
931
- xxhash==3.6.0
932
- # via datasets
933
- yarl==1.22.0
934
- # via aiohttp
935
- zipp==3.23.0
936
- # via
937
- # etils
938
- # importlib-metadata
939
- transformers==4.57.0
940
- # via
941
- # ads-gen (pyproject.toml)
942
- # dreamsim
943
- # peft
 
1
+ huggingface-hub==0.36.0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  torch==2.7.0
3
+ torchaudio==2.7.0
4
+ torchvision--0.22.0
5
+ transformers
6
+ diffusers==0.31.0
7
+ peft==0.15.2
8
+ opencv-python
9
+ protobuf
10
+ sentencepiece
11
+ gradio
12
+ jupyter
13
+ torchao
14
+ space