Update README.md
Browse files
README.md
CHANGED
|
@@ -102,7 +102,7 @@ pipeline = transformers.pipeline(
|
|
| 102 |
"text-generation",
|
| 103 |
model=model_id,
|
| 104 |
model_kwargs={"torch_dtype": torch.bfloat16},
|
| 105 |
-
revision="v2.0.0",
|
| 106 |
device_map="auto",
|
| 107 |
)
|
| 108 |
messages = [
|
|
|
|
| 102 |
"text-generation",
|
| 103 |
model=model_id,
|
| 104 |
model_kwargs={"torch_dtype": torch.bfloat16},
|
| 105 |
+
revision="v2.0.0", # Specify the revision here. Initial release is at "v2.0.0".
|
| 106 |
device_map="auto",
|
| 107 |
)
|
| 108 |
messages = [
|