hysts HF staff commited on
Commit
6eca2ab
1 Parent(s): a11d22c

Add paper info to the model card

Browse files
Files changed (1) hide show
  1. utils.py +11 -4
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'Test prompt: {test_prompt}\n' + f'![{image_path.stem}]({rel_path})\n'
 
 
 
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
- Base model: [{base_model}](https://huggingface.co/{base_model}).
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: