VideoTuna/pyproject.toml at main · testcoder-ui/VideoTuna · GitHub

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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
[tool.poetry]
name = "videotuna"
version = "0.1.0"
description = "Videotuna is a useful codebase for text-to-video applications"
authors = ["Yingqing He <yhebm@connect.ust.hk>", "Yazhou Xing <yxingag@connect.ust.hk>"]
readme = "README.md"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"

[tool.poetry.dependencies]
python = "^3.10"
deepspeed = "0.15.4"
av = "12.3.0"
beautifulsoup4 = "4.12.3"
colossalai = "0.3.6"
peft = "^0.12.0"
bitsandbytes = "^0.45.0"
decord = "0.6.0"
einops = "0.8.0"
fire = "0.6.0"
torch = "2.2.2"
ftfy = "6.2.3"
huggingface-hub = "0.24.6"
loguru = "0.7.2"
imwatermark = "0.0.2"
kornia = "0.7.3"
mmengine = "0.10.4"
omegaconf = "2.3.0"
opencv-python = "4.10.0.84"
packaging = "24.1"
pandas = "2.2.2"
pillow = "10.4.0"
pudb = "2024.1.2"
pytorch-lightning = "2.4.0"
pyyaml = "6.0.2"
rotary-embedding-torch = "0.6.5"
requests = "2.32.3"
safetensors = "0.4.4"
timm = "0.5.4"
torchvision = "0.17.2"
tqdm = "4.66.5"
transformers = "^4.44.0"
xformers = "0.0.25.post1"
imageio = "2.35.1"
imageio-ffmpeg = "0.5.1"
pyramid = "1.5"
wandb = "0.17.8"
scipy = "1.14.1"
beartype = "0.18.5"
moviepy = "1.0.3"
open-clip-torch = "2.12.0"
numpy = "==1.*"
diffusers = "^0.32.2"
torchsde = "0.2.6"
colorama = "0.4.6"
torch-optimi = "^0.2.1"
accelerate = "^0.33.0"
torchao = "0.8.0"
toml = "0.10.2"
hpsv2 = {git = "https://github.com/tgxs002/HPSv2.git"}
backports-tarfile = "^1.2.0"
swissarmytransformer = {git = "https://github.com/JingyeChen/SwissArmyTransformer"}

[tool.poetry.group.dev.dependencies]
black = "^24.0.0"
isort = "^5.12.0"
mypy = "^1.11.2"
pytest = "7.2.0"
pre-commit = "^4.0.0"
coverage = "^7.6.1"
ruff = "^0.6.8"

[tool.poetry.scripts]
install-flash-attn = 'scripts:install_flash_attn'
coverage-report = 'scripts:coverage_report'
format = 'scripts:code_format'
format-check = 'scripts:code_format_check'
lint = 'scripts:lint'
test = 'scripts:test'
type-check = 'scripts:type_check'
inference-cogvideo-i2v-diffusers = 'scripts:inference_cogvideo_i2v_diffusers'
inference-cogvideo-i2v-lora = 'scripts:inference_cogvideo_i2v_lora'
inference-cogvideo-lora = 'scripts:inference_cogvideo_lora'
inference-cogvideo-t2v-diffusers = 'scripts:inference_cogvideo_t2v_diffusers'
inference-cogvideox-15-5b-i2v = 'scripts:inference_cogvideox1_5_5b_i2v'
inference-cogvideox-15-5b-t2v = 'scripts:inference_cogvideox1_5_5b_t2v'
inference-dc-i2v-576x1024 = 'scripts:inference_dc_i2v_576x1024'
inference-flux-schnell = 'scripts:inference_flux_schnell'
inference-flux-dev = 'scripts:inference_flux_dev'
inference-flux-lora = 'scripts:inference_flux_lora'
inference-hunyuan = 'scripts:inference_hunyuan'
inference-mochi = 'scripts:inference_mochi'
inference-opensora-v10-16x256x256 = 'scripts:inference_opensora_v10_16x256x256'
inference-v2v-ms = 'scripts:inference_v2v_ms'
inference-vc1-i2v-320x512 = 'scripts:inference_vc1_i2v_320x512'
inference-vc1-t2v-576x1024 = 'scripts:inference_vc1_t2v_576x1024'
inference-vc2-t2v-320x512 = 'scripts:inference_vc2_t2v_320x512'
inference-vc2-t2v-320x512-lora = 'scripts:inference_vc2_t2v_320x512_lora'
train-cogvideox-i2v-lora = 'scripts:train_cogvideox_i2v_lora'
train-cogvideox-t2v-lora = 'scripts:train_cogvideox_t2v_lora'
train-dynamicrafter = 'scripts:train_dynamicrafter'
train-flux-lora = 'scripts:train_flux_lora'
train-opensorav10 = 'scripts:train_opensorav10'
train-videocrafter-lora = 'scripts:train_videocrafter_lora'
train-videocrafter-v2 = 'scripts:train_videocrafter_v2'

[tool.black]
line-length = 88
target-version = ['py310']
include = '\.pyi?$'

[tool.isort]
profile = "black"

[[tool.mypy.overrides]]
module = [
]
ignore_missing_imports = true

[tool.ruff]
select = ["E", "F", "C90"]
ignore = []