ford442 commited on
Commit
a91770e
·
verified ·
1 Parent(s): 5b4ec73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -8
app.py CHANGED
@@ -132,11 +132,6 @@ def save_image(img):
132
  img.save(unique_name,optimize=False,compress_level=0)
133
  return unique_name
134
 
135
- def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
136
- if randomize_seed:
137
- seed = random.randint(0, MAX_SEED)
138
- return seed
139
-
140
  def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp,denoise):
141
  filename= f'tst_A_{timestamp}.txt'
142
  with open(filename, "w") as f:
@@ -169,7 +164,7 @@ def generate_30(
169
  use_resolution_binning: bool = True,
170
  progress=gr.Progress(track_tqdm=True) # Add progress as a keyword argument
171
  ):
172
- seed = int(randomize_seed_fn(seed, randomize_seed))
173
  generator = torch.Generator(device='cuda').manual_seed(seed)
174
  options = {
175
  "prompt": [prompt],
@@ -209,7 +204,7 @@ def generate_60(
209
  use_resolution_binning: bool = True,
210
  progress=gr.Progress(track_tqdm=True) # Add progress as a keyword argument
211
  ):
212
- seed = int(randomize_seed_fn(seed, randomize_seed))
213
  generator = torch.Generator(device='cuda').manual_seed(seed)
214
  options = {
215
  "prompt": [prompt],
@@ -249,7 +244,7 @@ def generate_90(
249
  use_resolution_binning: bool = True,
250
  progress=gr.Progress(track_tqdm=True) # Add progress as a keyword argument
251
  ):
252
- seed = int(randomize_seed_fn(seed, randomize_seed))
253
  generator = torch.Generator(device='cuda').manual_seed(seed)
254
  options = {
255
  "prompt": [prompt],
 
132
  img.save(unique_name,optimize=False,compress_level=0)
133
  return unique_name
134
 
 
 
 
 
 
135
  def uploadNote(prompt,num_inference_steps,guidance_scale,timestamp,denoise):
136
  filename= f'tst_A_{timestamp}.txt'
137
  with open(filename, "w") as f:
 
164
  use_resolution_binning: bool = True,
165
  progress=gr.Progress(track_tqdm=True) # Add progress as a keyword argument
166
  ):
167
+ seed = random.randint(0, MAX_SEED)
168
  generator = torch.Generator(device='cuda').manual_seed(seed)
169
  options = {
170
  "prompt": [prompt],
 
204
  use_resolution_binning: bool = True,
205
  progress=gr.Progress(track_tqdm=True) # Add progress as a keyword argument
206
  ):
207
+ seed = random.randint(0, MAX_SEED)
208
  generator = torch.Generator(device='cuda').manual_seed(seed)
209
  options = {
210
  "prompt": [prompt],
 
244
  use_resolution_binning: bool = True,
245
  progress=gr.Progress(track_tqdm=True) # Add progress as a keyword argument
246
  ):
247
+ seed = random.randint(0, MAX_SEED)
248
  generator = torch.Generator(device='cuda').manual_seed(seed)
249
  options = {
250
  "prompt": [prompt],