Update app.py
Browse files
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"
|
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
|