Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
xuwx1
LightX2V
Commits
f6d4fc85
Commit
f6d4fc85
authored
Aug 05, 2025
by
PengGao
Committed by
GitHub
Aug 05, 2025
Browse files
style: add ruff isort (#183)
parent
878f5a48
Changes
133
Hide whitespace changes
Inline
Side-by-side
Showing
13 changed files
with
33 additions
and
29 deletions
+33
-29
scripts/server/check_status.py
scripts/server/check_status.py
+0
-1
scripts/server/post_enhancer.py
scripts/server/post_enhancer.py
+0
-1
scripts/server/post_multi_servers.py
scripts/server/post_multi_servers.py
+4
-3
scripts/server/post_multi_servers_i2v.py
scripts/server/post_multi_servers_i2v.py
+0
-1
scripts/server/post_multi_servers_t2v.py
scripts/server/post_multi_servers_t2v.py
+0
-1
scripts/server/post_vbench_i2v.py
scripts/server/post_vbench_i2v.py
+1
-0
scripts/server/stop_running_task.py
scripts/server/stop_running_task.py
+0
-1
tools/convert/converter.py
tools/convert/converter.py
+8
-6
tools/convert/trt_vae/convert_vae_trt_engine.py
tools/convert/trt_vae/convert_vae_trt_engine.py
+2
-2
tools/download_rife.py
tools/download_rife.py
+4
-3
tools/extract/convert_vigen_to_x2v_lora.py
tools/extract/convert_vigen_to_x2v_lora.py
+4
-4
tools/extract/lora_extractor.py
tools/extract/lora_extractor.py
+5
-3
tools/extract/lora_merger.py
tools/extract/lora_merger.py
+5
-3
No files found.
scripts/server/check_status.py
View file @
f6d4fc85
import
requests
from
loguru
import
logger
response
=
requests
.
get
(
"http://localhost:8000/v1/service/status"
)
logger
.
info
(
response
.
json
())
...
...
scripts/server/post_enhancer.py
View file @
f6d4fc85
import
requests
from
loguru
import
logger
url
=
"http://localhost:8000/v1/tasks/"
message
=
{
...
...
scripts/server/post_multi_servers.py
View file @
f6d4fc85
import
requests
from
loguru
import
logger
import
random
import
string
import
time
import
threading
import
time
from
datetime
import
datetime
import
requests
from
loguru
import
logger
from
tqdm
import
tqdm
...
...
scripts/server/post_multi_servers_i2v.py
View file @
f6d4fc85
from
loguru
import
logger
from
post_multi_servers
import
get_available_urls
,
process_tasks_async
if
__name__
==
"__main__"
:
urls
=
[
f
"http://localhost:
{
port
}
"
for
port
in
range
(
8000
,
8008
)]
img_prompts
=
{
...
...
scripts/server/post_multi_servers_t2v.py
View file @
f6d4fc85
from
loguru
import
logger
from
post_multi_servers
import
get_available_urls
,
process_tasks_async
if
__name__
==
"__main__"
:
urls
=
[
"http://localhost:8000"
,
"http://localhost:8001"
]
prompts
=
[
...
...
scripts/server/post_vbench_i2v.py
View file @
f6d4fc85
import
argparse
import
glob
import
os
from
loguru
import
logger
from
post_multi_servers
import
get_available_urls
,
process_tasks_async
...
...
scripts/server/stop_running_task.py
View file @
f6d4fc85
import
requests
from
loguru
import
logger
response
=
requests
.
get
(
"http://localhost:8000/v1/local/video/generate/stop_running_task"
)
logger
.
info
(
response
.
json
())
tools/convert/converter.py
View file @
f6d4fc85
import
os
import
re
import
argparse
import
gc
import
glob
import
json
import
argparse
import
os
import
re
import
shutil
from
collections
import
defaultdict
import
torch
from
safetensors
import
safe_open
,
torch
as
st
from
loguru
import
logger
from
tqdm
import
tqdm
from
collections
import
defaultdict
from
qtorch.quant
import
float_quantize
from
safetensors
import
safe_open
from
safetensors
import
torch
as
st
from
tqdm
import
tqdm
def
get_key_mapping_rules
(
direction
,
model_type
):
...
...
tools/convert/trt_vae/convert_vae_trt_engine.py
View file @
f6d4fc85
from
pathlib
import
Path
import
os
import
argparse
import
os
from
pathlib
import
Path
import
torch
from
loguru
import
logger
...
...
tools/download_rife.py
View file @
f6d4fc85
#!/usr/bin/env python3
# coding: utf-8
import
argparse
import
os
import
shutil
import
sys
import
requests
import
zipfile
import
shutil
import
argparse
from
pathlib
import
Path
import
requests
def
get_base_dir
():
"""Get project root directory"""
...
...
tools/extract/convert_vigen_to_x2v_lora.py
View file @
f6d4fc85
...
...
@@ -4,11 +4,11 @@
###
### ViGen-DiT Project Url: https://github.com/yl-1993/ViGen-DiT
###
import
torch
import
sys
import
os
from
safetensors.torch
import
save_file
from
safetensors.torch
import
load_file
import
sys
import
torch
from
safetensors.torch
import
load_file
,
save_file
if
len
(
sys
.
argv
)
!=
3
:
print
(
"用法: python convert_lora.py <输入文件> <输出文件.safetensors>"
)
...
...
tools/extract/lora_extractor.py
View file @
f6d4fc85
...
...
@@ -6,11 +6,13 @@ Extract LoRA weights from the difference between two models
"""
import
argparse
import
torch
from
safetensors
import
safe_open
,
torch
as
st
import
os
from
pathlib
import
Path
from
typing
import
Dict
,
Any
,
Optional
from
typing
import
Any
,
Dict
,
Optional
import
torch
from
safetensors
import
safe_open
from
safetensors
import
torch
as
st
from
tqdm
import
tqdm
...
...
tools/extract/lora_merger.py
View file @
f6d4fc85
...
...
@@ -6,11 +6,13 @@ Merge a source model with LoRA weights to create a new model
"""
import
argparse
import
torch
from
safetensors
import
safe_open
,
torch
as
st
import
os
from
pathlib
import
Path
from
typing
import
Dict
,
Any
,
Optional
from
typing
import
Any
,
Dict
,
Optional
import
torch
from
safetensors
import
safe_open
from
safetensors
import
torch
as
st
from
tqdm
import
tqdm
...
...
Prev
1
…
3
4
5
6
7
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment