generator seed fix for baseline

baseline.py

@@ -22,12 +22,11 @@ bg_negative = DEFAULT_OVERALL_NEGATIVE_PROMPT
 # Using dpm scheduler by default
 def run(prompt, scheduler_key='dpm_scheduler', bg_seed=1, num_inference_steps=20):
     print(f"prompt: {prompt}")
-    generator = torch.Generator(models.torch_device).manual_seed(bg_seed)
+    generator = torch.manual_seed(bg_seed)
     
     prompts = [prompt]
     input_embeddings = models.encode_prompts(prompts=prompts, tokenizer=tokenizer, text_encoder=text_encoder, negative_prompt=bg_negative)
 
-    generator = torch.manual_seed(1)  # Seed generator to create the inital latent noise
     latents = models.get_unscaled_latents(batch_size, unet.config.in_channels, height, width, generator, dtype)
 
     latents = latents * scheduler.init_noise_sigma