Update README
Browse files
README.md
CHANGED
|
@@ -200,7 +200,7 @@ If your cluster doesn’t allow internet access on compute nodes, you have two w
|
|
| 200 |
|
| 201 |
```bash
|
| 202 |
# On the front-end/login node (with internet):
|
| 203 |
-
python -c "from transformers import AutoModel; AutoModel.from_pretrained('sofieneb/histaug-
|
| 204 |
|
| 205 |
# On your compute job (no internet):
|
| 206 |
export HF_HUB_OFFLINE=1
|
|
@@ -213,7 +213,7 @@ This prevents unnecessary network calls and ensures `transformers` loads HistAug
|
|
| 213 |
|
| 214 |
```bash
|
| 215 |
# On the front-end/login node (with internet):
|
| 216 |
-
hf download sofieneb/histaug-
|
| 217 |
```
|
| 218 |
|
| 219 |
Then load the model from that directory in your script:
|
|
@@ -222,7 +222,7 @@ Then load the model from that directory in your script:
|
|
| 222 |
from transformers import AutoModel
|
| 223 |
|
| 224 |
cross_transformer = AutoModel.from_pretrained(
|
| 225 |
-
"./histaug-
|
| 226 |
trust_remote_code=True,
|
| 227 |
local_files_only=True
|
| 228 |
)
|
|
|
|
| 200 |
|
| 201 |
```bash
|
| 202 |
# On the front-end/login node (with internet):
|
| 203 |
+
python -c "from transformers import AutoModel; AutoModel.from_pretrained('sofieneb/histaug-conch', trust_remote_code=True)"
|
| 204 |
|
| 205 |
# On your compute job (no internet):
|
| 206 |
export HF_HUB_OFFLINE=1
|
|
|
|
| 213 |
|
| 214 |
```bash
|
| 215 |
# On the front-end/login node (with internet):
|
| 216 |
+
hf download sofieneb/histaug-conch --local-dir ./histaug-conch
|
| 217 |
```
|
| 218 |
|
| 219 |
Then load the model from that directory in your script:
|
|
|
|
| 222 |
from transformers import AutoModel
|
| 223 |
|
| 224 |
cross_transformer = AutoModel.from_pretrained(
|
| 225 |
+
"./histaug-conch", # local path instead of hub ID
|
| 226 |
trust_remote_code=True,
|
| 227 |
local_files_only=True
|
| 228 |
)
|