fix random number generator in session

singnet · Apr 26, 2024 · 5a432bb · 5a432bb
1 parent 07749b8
commit 5a432bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/multigen/sessions.py b/multigen/sessions.py
@@ -55,8 +55,8 @@ def gen_sess(self, add_count = 0, save_img=True,
             self.last_index = self.confg.count - 1
             self.last_conf = {**inputs}
             # TODO: multiple inputs?
-            # inputs['generator'] = torch.Generator().manual_seed(inputs['generator'])
-            inputs['generator'] = torch.cuda.manual_seed(inputs['generator'])
+            inputs['generator'] = torch.Generator().manual_seed(inputs['generator'])
+
             image = self.pipe.gen(inputs)
             if save_img:
                 self.last_img_name = self.get_last_file_prefix() + ".png"