diff --git a/examples/customized_exp/ppo_sentiment.py b/examples/customized_exp/ppo_sentiment.py index 6ee5c0d6..5b4a021f 100644 --- a/examples/customized_exp/ppo_sentiment.py +++ b/examples/customized_exp/ppo_sentiment.py @@ -26,7 +26,7 @@ class SentimentScoringInterface(model_api.ModelInterface): def __post_init__(self): - path = "/lustre/fw/pretrained/distilbert-base-uncased-finetuned-sst-2-english/" + path = "/path/to/model" self.score_model = ( transformers.AutoModelForSequenceClassification.from_pretrained(path).cuda() )