Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
2c49741
1
Parent(s):
5fa98d9
try fixes
Browse files
app.py
CHANGED
@@ -63,6 +63,7 @@ for idx, video, path in zip(
|
|
63 |
# pylint: disable-next=no-value-for-parameter
|
64 |
dfot = DFoTVideoPose.load_from_checkpoint(
|
65 |
checkpoint_path=download_pretrained("pretrained:DFoT_RE10K.ckpt"),
|
|
|
66 |
cfg=OmegaConf.load("config.yaml"),
|
67 |
).eval()
|
68 |
dfot.to("cuda")
|
@@ -651,7 +652,7 @@ with gr.Blocks(theme=gr.themes.Base(primary_hue="teal")) as demo:
|
|
651 |
demo2_fps = gr.Slider(
|
652 |
minimum=4,
|
653 |
maximum=20,
|
654 |
-
value=
|
655 |
step=1,
|
656 |
label="FPS",
|
657 |
info=f"A {LONG_LENGTH}-second video will be generated at this FPS; Decrease for faster generation; Increase for a smoother video",
|
|
|
63 |
# pylint: disable-next=no-value-for-parameter
|
64 |
dfot = DFoTVideoPose.load_from_checkpoint(
|
65 |
checkpoint_path=download_pretrained("pretrained:DFoT_RE10K.ckpt"),
|
66 |
+
map_location="cpu",
|
67 |
cfg=OmegaConf.load("config.yaml"),
|
68 |
).eval()
|
69 |
dfot.to("cuda")
|
|
|
652 |
demo2_fps = gr.Slider(
|
653 |
minimum=4,
|
654 |
maximum=20,
|
655 |
+
value=4,
|
656 |
step=1,
|
657 |
label="FPS",
|
658 |
info=f"A {LONG_LENGTH}-second video will be generated at this FPS; Decrease for faster generation; Increase for a smoother video",
|