kiwhansong commited on
Commit
2c49741
·
1 Parent(s): 5fa98d9
Files changed (1) hide show
  1. app.py +2 -1
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=8,
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",