CiaraRowles commited on
Commit
9f56d68
1 Parent(s): 3b76478

Update controlnet/callable_functions.py

Browse files
Files changed (1) hide show
  1. controlnet/callable_functions.py +4 -2
controlnet/callable_functions.py CHANGED
@@ -8,7 +8,7 @@ from controlnet.controlnetxs_appearance import StyleCodesModel
8
  from diffusers.models import UNet2DConditionModel
9
  from transformers import AutoProcessor, SiglipVisionModel
10
 
11
-
12
 
13
  def use_stylecode(model,image_path, prompt,negative_prompt, num_inference_steps, stylecode,seed=None,image=None):
14
  # Load and preprocess image
@@ -41,7 +41,9 @@ def use_stylecode(model,image_path, prompt,negative_prompt, num_inference_steps,
41
  if seed is not None and seed != -1:
42
  generator = torch.Generator(device="cuda").manual_seed(seed)
43
  else:
44
- generator = torch.Generator(device="cuda")
 
 
45
 
46
  # Run the image through the pipeline with the specified prompt
47
  output_images = pipe(
 
8
  from diffusers.models import UNet2DConditionModel
9
  from transformers import AutoProcessor, SiglipVisionModel
10
 
11
+ import random
12
 
13
  def use_stylecode(model,image_path, prompt,negative_prompt, num_inference_steps, stylecode,seed=None,image=None):
14
  # Load and preprocess image
 
41
  if seed is not None and seed != -1:
42
  generator = torch.Generator(device="cuda").manual_seed(seed)
43
  else:
44
+ random_seed = random.randint(0, 2**32 - 1)
45
+ print("using random seed ",random_seed)
46
+ generator = torch.Generator(device="cuda").manual_seed(random_seed)
47
 
48
  # Run the image through the pipeline with the specified prompt
49
  output_images = pipe(