showee commited on
Commit
c526f0f
·
1 Parent(s): 220e169

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -109,10 +109,12 @@ def error_str(error, title="Error"):
109
  {error}""" if error else ""
110
 
111
  def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt="", auto_prefix=False):
112
-
113
- generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
 
 
114
  prompt = f"{prefix} {prompt}" if auto_prefix else prompt
115
-
116
  try:
117
  if img is not None:
118
  return img_to_img(prompt, neg_prompt, img, strength, guidance, steps, width, height, generator), None
 
109
  {error}""" if error else ""
110
 
111
  def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt="", auto_prefix=False):
112
+ if torch.cuda.is_available():
113
+ generator = torch.Generator('cuda').manual_seed(seed) if seed != 0 else None
114
+ else:
115
+ generator = torch.Generator.manual_seed(seed) if seed != 0 else None
116
  prompt = f"{prefix} {prompt}" if auto_prefix else prompt
117
+
118
  try:
119
  if img is not None:
120
  return img_to_img(prompt, neg_prompt, img, strength, guidance, steps, width, height, generator), None