Spaces:
Running
Running
David Krajewski
commited on
Commit
·
108de9e
1
Parent(s):
7b22c42
added print cuda v
Browse files
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>""")
|