Spaces:

nnsohamnn
/

Conv_GPT

Sleeping

nnsohamnn commited on Mar 30, 2025

Commit

1f3d130

verified ·

1 Parent(s): 723b086

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ model = TransformerModel(
     num_heads=16,
     dropout=0.1
 )
-model.load_state_dict(torch.load("Conv_GPT.pth", map_location=torch.device('cpu')))
 model.eval()

     num_heads=16,
     dropout=0.1
 )
+model.load_state_dict(torch.load("Conv_GPT_finetuned_blended_skill.pth", map_location=torch.device('cpu')))
 model.eval()