switch to dtype instead of torch_dtype

dhia680 · web-flow · commit e7d03ad1d01d · 2025-08-27T23:55:00.000+01:00
Following this: huggingface#39782
diff --git a/docs/source/en/model_doc/apertus.md b/docs/source/en/model_doc/apertus.md
@@ -42,7 +42,7 @@ from transformers import pipeline
 pipeline = pipeline(
     task="text-generation",
     model="swiss-ai/Apertus-8B",
-    torch_dtype=torch.bfloat16,
+    dtype=torch.bfloat16,
     device=0
 )
 pipeline("Plants create energy through a process known as")
@@ -60,7 +60,7 @@ tokenizer = AutoTokenizer.from_pretrained(
 )
 model = AutoModelForCausalLM.from_pretrained(
     "swiss-ai/Apertus-8B",
-    torch_dtype=torch.bfloat16,
+    dtype=torch.bfloat16,
     device_map="auto",
     attn_implementation="sdpa"
 )

Original file line number	Diff line number	Diff line change
`@@ -42,7 +42,7 @@ from transformers import pipeline`
`42`	`42`	`pipeline = pipeline(`
`43`	`43`	`task="text-generation",`
`44`	`44`	`model="swiss-ai/Apertus-8B",`
`45`		`- torch_dtype=torch.bfloat16,`
	`45`	`+ dtype=torch.bfloat16,`
`46`	`46`	`device=0`
`47`	`47`	`)`
`48`	`48`	`pipeline("Plants create energy through a process known as")`
`@@ -60,7 +60,7 @@ tokenizer = AutoTokenizer.from_pretrained(`
`60`	`60`	`)`
`61`	`61`	`model = AutoModelForCausalLM.from_pretrained(`
`62`	`62`	`"swiss-ai/Apertus-8B",`
`63`		`- torch_dtype=torch.bfloat16,`
	`63`	`+ dtype=torch.bfloat16,`
`64`	`64`	`device_map="auto",`
`65`	`65`	`attn_implementation="sdpa"`
`66`	`66`	`)`