Grandediw commited on
Commit
1322782
·
verified ·
1 Parent(s): 087c7d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_repo, token=token)
18
  base_model = AutoModel.from_pretrained(model_repo, token=token)
19
 
20
  # Load the LoRA adapter weights
21
- lora_weights_path = "adapter_model.safetensors" # Ensure this file exists in the same directory
22
  lora_weights = load_file(lora_weights_path)
23
 
24
  # Apply LoRA weights to the base model
 
18
  base_model = AutoModel.from_pretrained(model_repo, token=token)
19
 
20
  # Load the LoRA adapter weights
21
+ lora_weights_path = "adapter_model.safetensors"
22
  lora_weights = load_file(lora_weights_path)
23
 
24
  # Apply LoRA weights to the base model