RowRed
/

CategoryRouter

category-classifier

cost-optimization

Model card Files Files and versions

RowRed commited on 13 days ago

Commit

b5935b0

·

verified ·

1 Parent(s): 4cc585e

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -187,7 +187,7 @@ class CategoryRouter(nn.Module):
 # Load
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-tokenizer = AutoTokenizer.from_pretrained("~~~ME~~~/CategoryRouter")
 model = CategoryRouter()
 model.load_state_dict(
     torch.load("pytorch_model.bin", map_location=device),

 # Load
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+tokenizer = AutoTokenizer.from_pretrained("RowRed/CategoryRouter")
 model = CategoryRouter()
 model.load_state_dict(
     torch.load("pytorch_model.bin", map_location=device),