Update README.md
Browse files
README.md
CHANGED
|
@@ -40,7 +40,7 @@ huggingface_hub.login(<ACCESS_TOKEN>)
|
|
| 40 |
from transformers import pipeline
|
| 41 |
|
| 42 |
generate_text = pipeline(
|
| 43 |
-
model="
|
| 44 |
torch_dtype="auto",
|
| 45 |
trust_remote_code=True,
|
| 46 |
device_map={"": "cuda:0"},
|
|
|
|
| 40 |
from transformers import pipeline
|
| 41 |
|
| 42 |
generate_text = pipeline(
|
| 43 |
+
model="SNUMPR/Protoss-c",
|
| 44 |
torch_dtype="auto",
|
| 45 |
trust_remote_code=True,
|
| 46 |
device_map={"": "cuda:0"},
|