Add paper info to the model card
Browse files
utils.py
CHANGED
@@ -29,7 +29,10 @@ def save_model_card(
|
|
29 |
if image_paths:
|
30 |
image_path = image_paths[-1]
|
31 |
rel_path = image_path.relative_to(save_dir)
|
32 |
-
image_str = f'
|
|
|
|
|
|
|
33 |
|
34 |
model_card = f'''---
|
35 |
license: creativeml-openrail-m
|
@@ -47,11 +50,15 @@ inference: false
|
|
47 |
|
48 |
# Tune-A-Video - {save_dir.name}
|
49 |
|
50 |
-
|
51 |
-
|
52 |
-
Training prompt: {training_prompt}
|
53 |
|
54 |
{image_str}
|
|
|
|
|
|
|
|
|
55 |
'''
|
56 |
|
57 |
with open(save_dir / 'README.md', 'w') as f:
|
|
|
29 |
if image_paths:
|
30 |
image_path = image_paths[-1]
|
31 |
rel_path = image_path.relative_to(save_dir)
|
32 |
+
image_str = f'''## Samples
|
33 |
+
Test prompt: {test_prompt}
|
34 |
+
|
35 |
+
![{image_path.stem}]({rel_path})'''
|
36 |
|
37 |
model_card = f'''---
|
38 |
license: creativeml-openrail-m
|
|
|
50 |
|
51 |
# Tune-A-Video - {save_dir.name}
|
52 |
|
53 |
+
## Model description
|
54 |
+
- Base model: [{base_model}](https://huggingface.co/{base_model})
|
55 |
+
- Training prompt: {training_prompt}
|
56 |
|
57 |
{image_str}
|
58 |
+
|
59 |
+
## Related papers:
|
60 |
+
- [Tune-A-Video](https://arxiv.org/abs/2212.11565): One-Shot Tuning of Image Diffusion Models for Text-to-Video Generation
|
61 |
+
- [Stable-Diffusion](https://arxiv.org/abs/2112.10752): High-Resolution Image Synthesis with Latent Diffusion Models
|
62 |
'''
|
63 |
|
64 |
with open(save_dir / 'README.md', 'w') as f:
|