dwb2023 commited on
Commit
f3873df
·
verified ·
1 Parent(s): e1bfa5b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -47,10 +47,10 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
47
 
48
  model_id = "google/paligemma-3b-ft-refcoco-seg-224"
49
  model = PaliGemmaForConditionalGeneration.from_pretrained(model_id).eval().to(device)
50
- # model = PeftModel.from_pretrained(model, adapter_model_id).to(device)
51
 
52
- model = model.merge_and_unload()
53
- model.save_pretrained("merged_adapters")
 
54
 
55
  processor = PaliGemmaProcessor.from_pretrained(model_id)
56
 
 
47
 
48
  model_id = "google/paligemma-3b-ft-refcoco-seg-224"
49
  model = PaliGemmaForConditionalGeneration.from_pretrained(model_id).eval().to(device)
 
50
 
51
+ # model = PeftModel.from_pretrained(model, adapter_model_id).to(device)
52
+ # model = model.merge_and_unload()
53
+ # model.save_pretrained("merged_adapters")
54
 
55
  processor = PaliGemmaProcessor.from_pretrained(model_id)
56