Update generate.py

generate.py

@@ -61,8 +61,8 @@ def generate(args):
             raise NotImplementedError(f"Unsupported prompt_extend_method: {args.prompt_extend_method}")
 
     cfg = WAN_CONFIGS[args.task]
-    logging.info(f"Generation job args: {args}")
-    logging.info(f"Generation model config: {cfg}")
+    print(f"Generation job args: {args}")
+    print(f"Generation model config: {cfg}")
 
     # Broadcast base seed across distributed workers
     if dist.is_initialized():
@@ -72,9 +72,10 @@ def generate(args):
 
     # Set prompt and task details
     if "t2v" in args.task or "t2i" in args.task:
+        print("tect to x ")
         if args.prompt is None:
             args.prompt = EXAMPLE_PROMPT[args.task]["prompt"]
-        logging.info(f"Input prompt: {args.prompt}")
+        print(f"Input prompt: {args.prompt}")
 
         if args.use_prompt_extend:
             logging.info("Extending prompt ...")