docs(src): :rocket: space

app.py

@@ -15,30 +15,32 @@ def generate_video(
     seed,
     num_inference_steps,
     guidance_scale,
-    flow_shift,
-    embedded_guidance_scale
+    flow_shift, # TODO: change to flow_shift 
+    embedded_guidance_scale # TODO: change to embedded_guidance_scale
 ):
     seed = None if seed == -1 else seed
     width, height = resolution.split("x")
     width, height = int(width), int(height)
 
     model = "hunyuanvideo-community/HunyuanVideo"
-    transformer_8bit = HunyuanVideoTransformer3DModel.from_pretrained(
+    transformer = HunyuanVideoTransformer3DModel.from_pretrained(
         model,
         subfolder="transformer",
         device_map="balanced",
-        torch_dtype=torch.float16,
+        torch_dtype=torch.float16, # TODO: change to bfloat16
     )
-    print(f"transformer_3bit device: {transformer_3bit.device}")
+    print(f"transformer device: {transformer.device}")
 
     # Cargar el pipeline
     pipeline = HunyuanVideoPipeline.from_pretrained(
         model,
-        transformer=transformer_8bit,
-        torch_dtype=torch.float16,
+        transformer=transformer,
+        torch_dtype=torch.float16, # TODO: change to bfloat16
         device_map="balanced",
     )
     print(f"pipeline device: {pipeline.device}")
+    # TODO: pipeline.vae.enable_tiling()
+    # TODO: pipeline.to("cuda")
     
     # Generar el video usando el pipeline
     video = pipeline(