patrickvonplaten commited on
Commit
26cd1e9
1 Parent(s): 03cbfd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -65,6 +65,9 @@ def convert_and_push(radio_model_names, input_model, ckpt_name, sd_version, toke
65
  new_checkpoint_path = "/".join(ckpt_path.split("/")[:-1] + ["model_safe.ckpt"])
66
  torch.save(tensors, new_checkpoint_path)
67
  ckpt_path = new_checkpoint_path
 
 
 
68
 
69
  # 2. Run the conversion script
70
  os.makedirs(model_id, exist_ok=True)
 
65
  new_checkpoint_path = "/".join(ckpt_path.split("/")[:-1] + ["model_safe.ckpt"])
66
  torch.save(tensors, new_checkpoint_path)
67
  ckpt_path = new_checkpoint_path
68
+ print("Converting ckpt_path", ckpt_path)
69
+
70
+ print(ckpt_path)
71
 
72
  # 2. Run the conversion script
73
  os.makedirs(model_id, exist_ok=True)