Update space

app.py

@@ -91,6 +91,7 @@ def trim_adapter_weights(model_path):
     key_to_trim = "lm_head.lora_B.default.weight"
     
     if key_to_trim in checkpoint:
+        print("Entered")
         original_size = checkpoint[key_to_trim].shape[0]
         expected_size = original_size - 1  # Removing last token
         
@@ -102,7 +103,7 @@ def trim_adapter_weights(model_path):
         trimmed_adapter_path = os.path.join(model_path, "adapter_model_trimmed.safetensors")
         save_file(checkpoint, trimmed_adapter_path)
         return trimmed_adapter_path
-    
+    print("Escaped from here hahaha")
     return model_path
 model_path=os.path.join(MODEL_1_PATH,"adapter_model.safetensors")
 trimmed_adapter_path = trim_adapter_weights(model_path)