Rooni commited on
Commit
7519957
·
verified ·
1 Parent(s): 405f02e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -41,7 +41,8 @@ def generate_cached_image(cache_key, prompt, width, height, seed, model_name):
41
  },
42
  "width": width,
43
  "height": height,
44
- "seed": seed
 
45
  }
46
 
47
  try:
 
41
  },
42
  "width": width,
43
  "height": height,
44
+ "seed": seed,
45
+ "generator": torch.Generator("cpu").manual_seed(seed)
46
  }
47
 
48
  try: