Sebukpor commited on
Commit
c276735
·
verified ·
1 Parent(s): 752d753

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ model, tokenizer = FastVisionModel.from_pretrained(
21
  # -----------------------------------------------------------------------------
22
 
23
  # Loading adapter WITHOUT get_peft_model()
24
- model = PeftModel.from_pretrained(model, "/content/fine_tuned_model")
25
  model = model.merge_and_unload() # Merge adapters into base model
26
  model.to("cuda")
27
  model.eval()
 
21
  # -----------------------------------------------------------------------------
22
 
23
  # Loading adapter WITHOUT get_peft_model()
24
+ model = PeftModel.from_pretrained(model, "fine_tuned_model")
25
  model = model.merge_and_unload() # Merge adapters into base model
26
  model.to("cuda")
27
  model.eval()