0.25 re-adding model_a.tie_weights()

app.py

@@ -66,6 +66,7 @@ def load_model_a(model_id):
             device_map="auto",
             trust_remote_code=True,
         )
+        model_a.tie_weights()
     except Exception as e:
         logging.error(f'{SPACER} Error: {e}, Traceback {traceback.format_exc()}')
     return gr.update(label=model_id)
@@ -82,6 +83,7 @@ def load_model_b(model_id):
             device_map="auto",
             trust_remote_code=True,
         )
+        model_b.tie_weights()
     except Exception as e:
         logging.error(f'{SPACER} Error: {e}, Traceback {traceback.format_exc()}')
     return gr.update(label=model_id)