Spaces:
Runtime error
Runtime error
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>""")
|