temporarily remove peft layer in the gradio code
Browse files
app.py
CHANGED
|
@@ -22,10 +22,10 @@ def load_model_once():
|
|
| 22 |
|
| 23 |
# Load your trained PEFT adapters
|
| 24 |
# Replace with your actual PEFT adapter path
|
| 25 |
-
model = PeftModel.from_pretrained(
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
)
|
| 29 |
return model, tokenizer
|
| 30 |
|
| 31 |
def summarize_text(text):
|
|
|
|
| 22 |
|
| 23 |
# Load your trained PEFT adapters
|
| 24 |
# Replace with your actual PEFT adapter path
|
| 25 |
+
# model = PeftModel.from_pretrained(
|
| 26 |
+
# base_model,
|
| 27 |
+
# "./path-to-your-adapter" # Upload your wandb artifact files here
|
| 28 |
+
# )
|
| 29 |
return model, tokenizer
|
| 30 |
|
| 31 |
def summarize_text(text):
|