Update worker_runpod.py

worker_runpod.py

@@ -18,6 +18,7 @@ from cogvideox.utils.lora_utils import merge_lora, unmerge_lora
 from cogvideox.utils.utils import get_image_to_video_latent, save_videos_grid, ASPECT_RATIO_512, get_closest_ratio, to_pil
 from huggingface_hub import HfApi, HfFolder
 
+tokenxf = os.getenv("HF_API_TOKEN")
 # Low GPU memory mode
 low_gpu_memory_mode = False
 
@@ -68,10 +69,10 @@ else:
 def generate(input):
     values = input["input"]
     prompt = values["prompt"]
-    negative_prompt = values.get("negative_prompt", "")
+    negative_prompt = values.get("negative_prompt", "blurry, blurred, blurry face")
     guidance_scale = values.get("guidance_scale", 6.0)
     seed = values.get("seed", 42)
-    num_inference_steps = values.get("num_inference_steps", 50)
+    num_inference_steps = values.get("num_inference_steps", 18)
     base_resolution = values.get("base_resolution", 512)
     
     video_length = values.get("video_length", 53)
@@ -120,18 +121,19 @@ def generate(input):
     video_path = os.path.join(save_path, f"{prefix}.mp4")
     save_videos_grid(sample, video_path, fps=fps)
 
-    # Upload final video to Hugging Face repository
-    #hf_api = HfApi()
-    #repo_id = values.get("repo_id", "your-username/your-repo")  # Set your HF repo
-    #hf_api.upload_file(
-    #    path_or_fileobj=video_path,
-    #    path_in_repo=f"{prefix}.mp4",
-    #    repo_id=repo_id,
-    #    repo_type="model"  # or "dataset" if using a dataset repo
-    #)
 
-    # Prepare output
-    #result_url = f"https://huggingface.co/{repo_id}/blob/main/{prefix}.mp4"
+    hf_api = HfApi()
+    repo_id = "meepmoo/h4h4jejdf"  # Set your HF repo
+    hf_api.upload_file(
+       path_or_fileobj=video_path,
+       path_in_repo=f"{prefix}.mp4",
+       repo_id=repo_id,
+       token=tokenxf,
+       repo_type="model"
+    )
+
+    Prepare output
+    result_url = f"https://huggingface.co/{repo_id}/blob/main/{prefix}.mp4"
     result_url = ""
     job_id = values.get("job_id", "default-job-id")  # For RunPod job tracking
     return {"jobId": job_id, "result": result_url, "status": "DONE"}