Create hunyuanvideo.py

setup.py

@@ -1,5 +1,5 @@
+from setuptools import find_packages, setup
 
-from setuptools import setup, find_packages
 
 install_requires = ["ruff"]
 

src/poc_style_bot/__init__.py

@@ -1,4 +1,3 @@
-
 __version__ = "0.1.0"
 
 from .geometry import circle_area, circle_circumference

src/poc_style_bot/geometry.py

@@ -1,5 +1,9 @@
-
 import math
 
-def circle_area(radius): return math.pi * radius ** 2
-def circle_circumference(radius): return 2 * math.pi * radius
+
+def circle_area(radius):
+    return math.pi * radius**2
+
+
+def circle_circumference(radius):
+    return 2 * math.pi * radius

src/poc_style_bot/hunyuanvideo.py

@@ -0,0 +1,20 @@
+import torch
+
+from diffusers import HunyuanVideoPipeline, HunyuanVideoTransformer3DModel
+from diffusers.utils import export_to_video
+
+
+model_id = "tencent/HunyuanVideo"
+transformer = HunyuanVideoTransformer3DModel.from_pretrained(
+    model_id, subfolder="transformer", torch_dtype=torch.bfloat16, revision="refs/pr/18"
+)
+pipe = HunyuanVideoPipeline.from_pretrained(
+    model_id, transformer=transformer, torch_dtype=torch.float16, revision="refs/pr/18"
+)
+pipe.vae.enable_tiling()
+pipe.to("cuda")
+
+output = pipe(
+    prompt="A cat walks on the grass, realistic", height=320, width=512, num_frames=61, num_inference_steps=30
+).frames[0]
+export_to_video(output, "output_hunyuan.mp4", fps=15)

tests/test_geometry.py

@@ -1,37 +1,39 @@
-import unittest
 import math
+import unittest
+
 from poc_style_bot import circle_area, circle_circumference
 
+
 class TestCircleFunctions(unittest.TestCase):
     def test_circle_area(self):
         # Test area with a unit radius
         self.assertAlmostEqual(circle_area(1), math.pi)
-        
+
         # Test area with zero radius
         self.assertAlmostEqual(circle_area(0), 0.0)
-        
+
         # Test area with a fractional radius
         radius = 2.5
-        expected = math.pi * radius ** 2
+        expected = math.pi * radius**2
         self.assertAlmostEqual(circle_area(radius), expected)
-        
+
         # Test with a large radius
         radius = 100
-        expected = math.pi * (100 ** 2)
+        expected = math.pi * (100**2)
         self.assertAlmostEqual(circle_area(radius), expected)
 
     def test_circle_circumference(self):
         # Test circumference with a unit radius
         self.assertAlmostEqual(circle_circumference(1), 2 * math.pi)
-        
+
         # Test circumference with zero radius
         self.assertAlmostEqual(circle_circumference(0), 0.0)
-        
+
         # Test circumference with a fractional radius
         radius = 2.5
         expected = 2 * math.pi * radius
         self.assertAlmostEqual(circle_circumference(radius), expected)
-        
+
         # Test with a large radius
         radius = 100
         expected = 2 * math.pi * 100