Update README.md

README.md

@@ -43,7 +43,7 @@ from diffusers.utils import export_to_video
 # Load the pipeline
 pipe = Kandinsky5T2VPipeline.from_pretrained(
     "ai-forever/Kandinsky-5.0-T2V-Lite-sft-5s-Diffusers", 
-    torch_dtype=torch.float16
+    torch_dtype=torch.bfloat16
 )
 pipe = pipe.to("cuda")
 
@@ -73,25 +73,25 @@ from diffusers import Kandinsky5T2VPipeline
 # SFT 5s model (highest quality)
 pipe_sft = Kandinsky5T2VPipeline.from_pretrained(
     "ai-forever/Kandinsky-5.0-T2V-Lite-sft-5s-Diffusers",
-    torch_dtype=torch.float16
+    torch_dtype=torch.bfloat16
 )
 
 # Distilled 16-step model (fastest)
 pipe_distill = Kandinsky5T2VPipeline.from_pretrained(
     "ai-forever/Kandinsky-5.0-T2V-Lite-distilled16steps-5s-Diffusers", 
-    torch_dtype=torch.float16
+    torch_dtype=torch.bfloat16
 )
 
 # No-CFG model (balanced speed/quality)
 pipe_nocfg = Kandinsky5T2VPipeline.from_pretrained(
     "ai-forever/Kandinsky-5.0-T2V-Lite-nocfg-5s-Diffusers",
-    torch_dtype=torch.float16
+    torch_dtype=torch.bfloat16
 )
 
 # Pretrain model (most diverse)
 pipe_pretrain = Kandinsky5T2VPipeline.from_pretrained(
     "ai-forever/Kandinsky-5.0-T2V-Lite-pretrain-5s-Diffusers",
-    torch_dtype=torch.float16
+    torch_dtype=torch.bfloat16
 )
 ```