Update utils/keyframe_utils.py

utils/keyframe_utils.py

@@ -16,7 +16,7 @@ CACHE_DIR.mkdir(exist_ok=True)
 LOG_PATH = Path("prompt_log.jsonl")
 
 pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+pipe = pipe.to("cpu")
 
 def generate_keyframe_prompt(segment):
     segment_id = segment.get("segment_id")
@@ -91,4 +91,4 @@ def generate_all_keyframe_images(script_data, output_dir="keyframes"):
     with open("all_prompts_output.json", "w", encoding="utf-8") as f:
         json.dump(keyframe_outputs, f, ensure_ascii=False, indent=2)
 
-    return keyframe_outputs
+    return keyframe_outputs