Spaces:
Build error
Build error
fix: method call.
Browse files- convert.py +3 -1
convert.py
CHANGED
@@ -59,7 +59,9 @@ def initialize_tf_models(text_encoder_weights: str, unet_weights: str):
|
|
59 |
|
60 |
def run_conversion(text_encoder_weights: str = None, unet_weights: str = None):
|
61 |
pt_text_encoder, pt_vae, pt_unet, pt_safety_checker = initialize_pt_models()
|
62 |
-
tf_sd_model, tf_text_encoder, tf_vae, tf_unet = initialize_tf_models(
|
|
|
|
|
63 |
print("Pre-trained model weights downloaded.")
|
64 |
|
65 |
if text_encoder_weights is not None:
|
|
|
59 |
|
60 |
def run_conversion(text_encoder_weights: str = None, unet_weights: str = None):
|
61 |
pt_text_encoder, pt_vae, pt_unet, pt_safety_checker = initialize_pt_models()
|
62 |
+
tf_sd_model, tf_text_encoder, tf_vae, tf_unet = initialize_tf_models(
|
63 |
+
text_encoder_weights, unet_weights
|
64 |
+
)
|
65 |
print("Pre-trained model weights downloaded.")
|
66 |
|
67 |
if text_encoder_weights is not None:
|