David Krajewski commited on
Commit
108de9e
·
1 Parent(s): 7b22c42

added print cuda v

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -92,6 +92,8 @@ def download_models(ckpts_path):
92
  snapshot_download(repo_id="MyNiuuu/MOFA-Video-Traj", local_dir=ckpts_path, cache_dir=ckpts_path, allow_patterns=["ckpts/controlnet/*"])
93
  except (Exception, BaseException) as error:
94
  print(error)
 
 
95
 
96
  def init_models(pretrained_model_name_or_path, resume_from_checkpoint, weight_dtype, device='cuda', enable_xformers_memory_efficient_attention=False, allow_tf32=False):
97
 
@@ -645,6 +647,8 @@ class Drag:
645
 
646
 
647
  with gr.Blocks() as demo:
 
 
648
  gr.Markdown("""<h1 align="center">MOFA-Video</h1><br>""")
649
 
650
  gr.Markdown("""Official Gradio Demo for <a href='https://myniuuu.github.io/MOFA_Video'><b>MOFA-Video: Controllable Image Animation via Generative Motion Field Adaptions in Frozen Image-to-Video Diffusion Model</b></a>.<br>""")
 
92
  snapshot_download(repo_id="MyNiuuu/MOFA-Video-Traj", local_dir=ckpts_path, cache_dir=ckpts_path, allow_patterns=["ckpts/controlnet/*"])
93
  except (Exception, BaseException) as error:
94
  print(error)
95
+
96
+ print("models downloaded!")
97
 
98
  def init_models(pretrained_model_name_or_path, resume_from_checkpoint, weight_dtype, device='cuda', enable_xformers_memory_efficient_attention=False, allow_tf32=False):
99
 
 
647
 
648
 
649
  with gr.Blocks() as demo:
650
+ print(f"Current CUDA version: {torch.version.cuda}")
651
+
652
  gr.Markdown("""<h1 align="center">MOFA-Video</h1><br>""")
653
 
654
  gr.Markdown("""Official Gradio Demo for <a href='https://myniuuu.github.io/MOFA_Video'><b>MOFA-Video: Controllable Image Animation via Generative Motion Field Adaptions in Frozen Image-to-Video Diffusion Model</b></a>.<br>""")