Update README.md
Browse files
README.md
CHANGED
|
@@ -136,7 +136,7 @@ Once, setup you can proceed to run the model by running the snippet below:
|
|
| 136 |
```py
|
| 137 |
from transformers import pipeline
|
| 138 |
import torch
|
| 139 |
-
model_id = "EpistemeAI/VibeCoder-20b-
|
| 140 |
pipe = pipeline(
|
| 141 |
"text-generation",
|
| 142 |
model=model_id,
|
|
@@ -168,7 +168,7 @@ except ValueError:
|
|
| 168 |
|
| 169 |
# Hub Model configuration. https://huggingface.co/models
|
| 170 |
hub = {
|
| 171 |
-
'HF_MODEL_ID':'EpistemeAI/VibeCoder-20b-
|
| 172 |
'SM_NUM_GPUS': json.dumps(1)
|
| 173 |
}
|
| 174 |
|
|
|
|
| 136 |
```py
|
| 137 |
from transformers import pipeline
|
| 138 |
import torch
|
| 139 |
+
model_id = "EpistemeAI/VibeCoder-20b-RL1.0"
|
| 140 |
pipe = pipeline(
|
| 141 |
"text-generation",
|
| 142 |
model=model_id,
|
|
|
|
| 168 |
|
| 169 |
# Hub Model configuration. https://huggingface.co/models
|
| 170 |
hub = {
|
| 171 |
+
'HF_MODEL_ID':'EpistemeAI/VibeCoder-20b-RL1.0',
|
| 172 |
'SM_NUM_GPUS': json.dumps(1)
|
| 173 |
}
|
| 174 |
|