diff --git a/setup.py b/setup.py index 791eed6..135b0ed 100644 --- a/setup.py +++ b/setup.py @@ -1,5 +1,5 @@ +from setuptools import find_packages, setup -from setuptools import setup, find_packages install_requires = ["ruff"] diff --git a/src/poc_style_bot/__init__.py b/src/poc_style_bot/__init__.py index 965811c..18c5202 100644 --- a/src/poc_style_bot/__init__.py +++ b/src/poc_style_bot/__init__.py @@ -1,4 +1,3 @@ - __version__ = "0.1.0" from .geometry import circle_area, circle_circumference diff --git a/src/poc_style_bot/geometry.py b/src/poc_style_bot/geometry.py index 4345751..dbf8588 100644 --- a/src/poc_style_bot/geometry.py +++ b/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 diff --git a/src/poc_style_bot/hunyuanvideo.py b/src/poc_style_bot/hunyuanvideo.py new file mode 100644 index 0000000..e254f5f --- /dev/null +++ b/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) diff --git a/tests/test_geometry.py b/tests/test_geometry.py index abeb257..91c0903 100644 --- a/tests/test_geometry.py +++ b/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