Fix passing step to wandb.

hypertrain.py

@@ -384,7 +384,8 @@ for input_ids, labels in t:
 
     if curr_step % train_config["eval_every"] == 0 and curr_step != 0:
         for prompt in prompts:
-            sampled = sample(prompt, 500, 3, hypernetwork=hypernetwork)
+            sampled = sample(prompt, 500, 3, hypernetwork=hypernetwork,
+                             step=step)
             print(f"PROMPT:\n{prompt}")
             report_console(sampled)
             sample_data = sample_data + sampled