slymnyldrm commited on
Commit
742d313
1 Parent(s): d484d65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,9 +12,10 @@ pipe = StableDiffusionPipeline.from_pretrained(model_path, use_safetensors=True,
12
  pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
13
 
14
  g_cuda = torch.Generator()
 
15
  g_cuda.manual_seed(g_seed)
16
 
17
- def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_inference_steps=50, guidance_scale=7.5, g_seed=52362):
18
  with torch.inference_mode():
19
  return pipe(
20
  prompt, height=int(height), width=int(width),
 
12
  pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
13
 
14
  g_cuda = torch.Generator()
15
+ g_seed=52362
16
  g_cuda.manual_seed(g_seed)
17
 
18
+ def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_inference_steps=50, guidance_scale=7.5):
19
  with torch.inference_mode():
20
  return pipe(
21
  prompt, height=int(height), width=int(width),