Priyanka6 commited on
Commit
eb72a06
Β·
1 Parent(s): f214076

Update space

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -81,7 +81,7 @@ def trim_adapter_weights(model_path):
81
  Trims the last token from the adapter's lm_head.lora_B.default.weight
82
  if there is a mismatch with the base model.
83
  """
84
- model_path = "Priyanka6/fine-tuning-inference/tree/main/adapter_model.safetensors"
85
  # if not os.path.exists(model_path):
86
  # raise FileNotFoundError(f"Adapter file not found: {model_path}")
87
 
 
81
  Trims the last token from the adapter's lm_head.lora_B.default.weight
82
  if there is a mismatch with the base model.
83
  """
84
+ model_path = "./adapter_model.safetensors"
85
  # if not os.path.exists(model_path):
86
  # raise FileNotFoundError(f"Adapter file not found: {model_path}")
87