Upload 5 files

hyvideo/constants.py

@@ -17,9 +17,9 @@ __all__ = [
 ]
 
 PRECISION_TO_TYPE = {
-    "fp32": torch.float32,
-    "fp16": torch.float16,
-    "bf16": torch.bfloat16,
+    'fp32': torch.float32,
+    'fp16': torch.float16,
+    'bf16': torch.bfloat16,
 }
 
 # =================== Constant Values =====================

hyvideo/prompt_rewrite.py

@@ -25,7 +25,7 @@ You are a large language model specialized in rewriting video descriptions. Your
 
 1. If the input is in Chinese, translate the entire description to English. 
 
-2. If the input is just one or two words describing an object or person, provide a brief, simple description focusing on basic visual characteristics. Limit the description to 1-2 short sentences.
+2. To generate high-quality visual scenes with aesthetic appeal, it is necessary to carefully depict each visual element to create a unique aesthetic.
 
 3. If the input does not include style, lighting, atmosphere, you can make reasonable associations.
 
@@ -48,4 +48,4 @@ ori_prompt = "一只小狗在草地上奔跑。"
 normal_prompt = get_rewrite_prompt(ori_prompt, mode="Normal")
 master_prompt = get_rewrite_prompt(ori_prompt, mode="Master")
 
-# Then you can use the normal_prompt or master_prompt to access the hunyuan-large rewrite model to get the final prompt.
+# Then you can use the normal_prompt or master_prompt to access the hunyuan-large rewrite model to get the final prompt.