Update demo_util.py

demo_util.py

@@ -57,6 +57,14 @@ def sample_fn(generator,
               randomize_temperature=2.0,
               num_sample_steps=8,
               device="cuda"):
+    print(f"generator:{generator}")
+    print(f"tokenizer:{tokenizer}")
+    print(f"labels:{labels}")
+    print(f"guidance_scale:{guidance_scale}")
+    print(f"randomize_temperature:{randomize_temperature}")
+    print(f"num_sample_steps:{num_sample_steps}")
+    print(f"device:{device}")
+    
     generator.eval()
     tokenizer.eval()
     if labels is None: