Update README.md
Browse files
README.md
CHANGED
|
@@ -76,7 +76,7 @@ class RouterMLP(nn.Module):
|
|
| 76 |
return self.cls(x).squeeze(-1)
|
| 77 |
|
| 78 |
# 2. Download and Load the Checkpoint
|
| 79 |
-
repo_id = "
|
| 80 |
model_path = hf_hub_download(repo_id=repo_id, filename="model.pt")
|
| 81 |
|
| 82 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 76 |
return self.cls(x).squeeze(-1)
|
| 77 |
|
| 78 |
# 2. Download and Load the Checkpoint
|
| 79 |
+
repo_id = "StevenMup2004/DynaRoute" # <-- Replace with your repo name
|
| 80 |
model_path = hf_hub_download(repo_id=repo_id, filename="model.pt")
|
| 81 |
|
| 82 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|