Update README.md
Browse files
README.md
CHANGED
|
@@ -27,8 +27,7 @@ pipe = pipeline(
|
|
| 27 |
"text-generation",
|
| 28 |
model="Bruce1489/Llama3.2-docker-command-v2",
|
| 29 |
torch_dtype=torch.bfloat16,
|
| 30 |
-
device_map="auto"
|
| 31 |
-
# tokenizer = tokenizer
|
| 32 |
)
|
| 33 |
messages = [
|
| 34 |
{"role": "system", "content": "translate this sentence in docker command"},
|
|
|
|
| 27 |
"text-generation",
|
| 28 |
model="Bruce1489/Llama3.2-docker-command-v2",
|
| 29 |
torch_dtype=torch.bfloat16,
|
| 30 |
+
device_map="auto"
|
|
|
|
| 31 |
)
|
| 32 |
messages = [
|
| 33 |
{"role": "system", "content": "translate this sentence in docker command"},
|