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",
|