Update t2v_test.py

t2v_test.py

@@ -44,12 +44,14 @@ class Text2Video():
         noise_shape = [batch_size, channels, frames, h, w]
 
         #prompts = batch_size * [""]
+        prompt = prompt+", professional, 4k, highly detailed"
+        neg_prompts = "graphite, impressionist, noisy, blurry, soft, deformed, ugly"
         text_emb = model.get_learned_conditioning([prompt])
 
         cond = {"c_crossattn": [text_emb], "fps": fps}
         
         ## inference
-        batch_samples = batch_ddim_sampling(model, cond, noise_shape, n_samples=1, ddim_steps=steps, ddim_eta=eta, cfg_scale=cfg_scale)
+        batch_samples = batch_ddim_sampling(model, cond, noise_shape, n_samples=1, ddim_steps=steps, ddim_eta=eta, cfg_scale=cfg_scale, neg_prompts=neg_prompts)
         ## b,samples,c,t,h,w
         prompt_str = prompt.replace("/", "_slash_") if "/" in prompt else prompt
         prompt_str = prompt_str.replace(" ", "_") if " " in prompt else prompt_str