reydeuss commited on
Commit
4d52ef0
·
verified ·
1 Parent(s): 59f31f8

temporarily remove peft layer in the gradio code

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- base_model,
27
- "./path-to-your-adapter" # Upload your wandb artifact files here
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):