Spaces:
Build error
Build error
fix: save_model_card call.
Browse files
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
|