sayakpaul HF staff commited on
Commit
30552b4
1 Parent(s): 6c04d23

fix: method call.

Browse files
Files changed (1) hide show
  1. 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: