Bruce1489
/

Llama3.2-docker-command-v2

Text Generation

text-generation-inference

Model card Files Files and versions Community

Bruce1489 commited on 11 days ago

Commit

538255a

·

verified ·

1 Parent(s): d498f48

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -27,8 +27,7 @@ pipe = pipeline(
     "text-generation",
     model="Bruce1489/Llama3.2-docker-command-v2",
     torch_dtype=torch.bfloat16,
-    device_map="auto",
-    # tokenizer = tokenizer
 )
 messages = [
     {"role": "system", "content": "translate this sentence in docker command"},

     "text-generation",
     model="Bruce1489/Llama3.2-docker-command-v2",
     torch_dtype=torch.bfloat16,
+    device_map="auto"
 )
 messages = [
     {"role": "system", "content": "translate this sentence in docker command"},