sayakpaul HF staff commited on
Commit
0554219
1 Parent(s): 9435d99

fix: save_model_card call.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,6 +9,7 @@ This Space lets you convert KerasCV Stable Diffusion weights to a format compati
9
 
10
  ## Notes (important)
11
 
 
12
  * Only Stable Diffusion (v1) is supported as of now. In particular this checkpoint: [`"CompVis/stable-diffusion-v1-4"`](https://huggingface.co/CompVis/stable-diffusion-v1-4).
13
  * Only the text encoder and the UNet parameters converted since only these two elements are generally fine-tuned.
14
  * [This Colab Notebook](https://colab.research.google.com/drive/1RYY077IQbAJldg8FkK8HSEpNILKHEwLb?usp=sharing) was used to develop the conversion utilities initially.
@@ -33,7 +34,6 @@ def run(hf_token, text_encoder_weights, unet_weights, repo_prefix):
33
  base_model=PRETRAINED_CKPT,
34
  repo_folder=output_path,
35
  weight_paths=[text_encoder_weights, unet_weights],
36
- repo_prefix=repo_prefix,
37
  )
38
  push_str = push_to_hub(hf_token, output_path, repo_prefix)
39
  return push_str
 
9
 
10
  ## Notes (important)
11
 
12
+ * The Space downloads a couple of pre-trained weights and runs a dummy inference. Depending, on the machine type, the enture process can take anywhere between 2 - 5 minutes.
13
  * Only Stable Diffusion (v1) is supported as of now. In particular this checkpoint: [`"CompVis/stable-diffusion-v1-4"`](https://huggingface.co/CompVis/stable-diffusion-v1-4).
14
  * Only the text encoder and the UNet parameters converted since only these two elements are generally fine-tuned.
15
  * [This Colab Notebook](https://colab.research.google.com/drive/1RYY077IQbAJldg8FkK8HSEpNILKHEwLb?usp=sharing) was used to develop the conversion utilities initially.
 
34
  base_model=PRETRAINED_CKPT,
35
  repo_folder=output_path,
36
  weight_paths=[text_encoder_weights, unet_weights],
 
37
  )
38
  push_str = push_to_hub(hf_token, output_path, repo_prefix)
39
  return push_str