Update controlnet/callable_functions.py

controlnet/callable_functions.py

@@ -8,7 +8,7 @@ from controlnet.controlnetxs_appearance import StyleCodesModel
 from diffusers.models import UNet2DConditionModel
 from transformers import AutoProcessor, SiglipVisionModel
 
-
+import random
 
 def use_stylecode(model,image_path, prompt,negative_prompt, num_inference_steps, stylecode,seed=None,image=None):
     # Load and preprocess image
@@ -41,7 +41,9 @@ def use_stylecode(model,image_path, prompt,negative_prompt, num_inference_steps,
     if seed is not None and seed != -1:
         generator = torch.Generator(device="cuda").manual_seed(seed)
     else:
-        generator = torch.Generator(device="cuda")
+        random_seed = random.randint(0, 2**32 - 1)
+        print("using random seed ",random_seed)
+        generator = torch.Generator(device="cuda").manual_seed(random_seed)
 
     # Run the image through the pipeline with the specified prompt
     output_images = pipe(