Spaces:
Running
Running
nalin0503
commited on
Commit
·
ca501e3
1
Parent(s):
d40dbb3
bug fixes
Browse files- run_morphing.py +7 -7
run_morphing.py
CHANGED
@@ -1,5 +1,3 @@
|
|
1 |
-
# T.B.C.
|
2 |
-
|
3 |
import os
|
4 |
import sys
|
5 |
import time
|
@@ -95,8 +93,8 @@ def parse_arguments():
|
|
95 |
)
|
96 |
parser.add_argument(
|
97 |
"--use_lcm", action="store_true",
|
98 |
-
|
99 |
-
|
100 |
|
101 |
# ------------------- FILM ARGS -------------------
|
102 |
parser.add_argument(
|
@@ -125,9 +123,9 @@ def parse_arguments():
|
|
125 |
def run_diffmorpher(args):
|
126 |
"""
|
127 |
Calls DiffMorpher's main.py via subprocess using the CLI arguments.
|
128 |
-
Expects `
|
129 |
"""
|
130 |
-
diffmorpher_script = os.path.join("
|
131 |
|
132 |
cmd = [
|
133 |
sys.executable, diffmorpher_script,
|
@@ -157,6 +155,9 @@ def run_diffmorpher(args):
|
|
157 |
cmd.append("--save_inter")
|
158 |
if args.no_lora:
|
159 |
cmd.append("--no_lora")
|
|
|
|
|
|
|
160 |
|
161 |
print("[INFO] Running DiffMorpher with command:")
|
162 |
print(" ".join(cmd))
|
@@ -166,7 +167,6 @@ def run_diffmorpher(args):
|
|
166 |
end = time.time()
|
167 |
print(f"[INFO] DiffMorpher completed in {end - start:.2f} seconds.")
|
168 |
|
169 |
-
|
170 |
def create_simple_video_from_keyframes(keyframes_folder, output_folder, fps=40):
|
171 |
"""
|
172 |
If the user does NOT want FILM, we still make a basic video from keyframes.
|
|
|
|
|
|
|
1 |
import os
|
2 |
import sys
|
3 |
import time
|
|
|
93 |
)
|
94 |
parser.add_argument(
|
95 |
"--use_lcm", action="store_true",
|
96 |
+
help="Enable LCM-LoRA acceleration for faster sampling"
|
97 |
+
)
|
98 |
|
99 |
# ------------------- FILM ARGS -------------------
|
100 |
parser.add_argument(
|
|
|
123 |
def run_diffmorpher(args):
|
124 |
"""
|
125 |
Calls DiffMorpher's main.py via subprocess using the CLI arguments.
|
126 |
+
Expects `Image-Morpher/` to be a submodule in the current repo.
|
127 |
"""
|
128 |
+
diffmorpher_script = os.path.join("Image-Morpher", "main.py")
|
129 |
|
130 |
cmd = [
|
131 |
sys.executable, diffmorpher_script,
|
|
|
155 |
cmd.append("--save_inter")
|
156 |
if args.no_lora:
|
157 |
cmd.append("--no_lora")
|
158 |
+
# ---- Add LCM-LoRA flag if set ----
|
159 |
+
if args.use_lcm:
|
160 |
+
cmd.append("--use_lcm")
|
161 |
|
162 |
print("[INFO] Running DiffMorpher with command:")
|
163 |
print(" ".join(cmd))
|
|
|
167 |
end = time.time()
|
168 |
print(f"[INFO] DiffMorpher completed in {end - start:.2f} seconds.")
|
169 |
|
|
|
170 |
def create_simple_video_from_keyframes(keyframes_folder, output_folder, fps=40):
|
171 |
"""
|
172 |
If the user does NOT want FILM, we still make a basic video from keyframes.
|