jjuun commited on
Commit
cc3479e
·
1 Parent(s): 9e6db6c

fix seed code to set user num

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ def model_load():
22
  def sdxl_process(seed, prompt, additional_prompt, negative_prompt, num_steps, guidance_scale):
23
  pipe = model_load()
24
  generator = torch.Generator("cuda")
25
- # generator.manual_seed(seed)
26
 
27
  special_prompt = 'jjj, scratch art style'
28
  prompt = f'{special_prompt}, {prompt}, with a black background'
 
22
  def sdxl_process(seed, prompt, additional_prompt, negative_prompt, num_steps, guidance_scale):
23
  pipe = model_load()
24
  generator = torch.Generator("cuda")
25
+ generator.manual_seed(int(seed))
26
 
27
  special_prompt = 'jjj, scratch art style'
28
  prompt = f'{special_prompt}, {prompt}, with a black background'