Update controlnet/callable_functions.py

controlnet/callable_functions.py

@@ -17,7 +17,7 @@ def use_stylecode(model,image_path, prompt,negative_prompt, num_inference_steps,
     stylecodes_model = StyleCodesModel.from_unet(unet, size_ratio=1.0).to(dtype=torch.float16, device="cuda")
 
 
-    print("running prompt = ",prompt, " negative_prompt = ",negative_prompt, " with code ", stylecode)
+    print("running prompt = ",prompt, " negative_prompt = ",negative_prompt, " with code ", stylecode, " and seed ",seed)
     stylecodes_model.load_model(model)
 
     pipe = StableDiffusionPipelineXSv2.from_pretrained(
@@ -41,7 +41,7 @@ def use_stylecode(model,image_path, prompt,negative_prompt, num_inference_steps,
     if seed is not None and not -1:
         generator = torch.Generator(device="cuda").manual_seed(seed)
     else:
-        generator = None
+        generator = torch.Generator(device="cuda")
 
     # Run the image through the pipeline with the specified prompt
     output_images = pipe(