Round Error Solved
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ torch.cuda.empty_cache()
|
|
| 26 |
def generate_image(prompt, seed=42, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT, randomize_seed=False, num_inference_steps=2, progress=gr.Progress(track_tqdm=True)):
|
| 27 |
if randomize_seed:
|
| 28 |
seed = random.randint(0, MAX_SEED)
|
| 29 |
-
generator = torch.Generator().manual_seed(seed)
|
| 30 |
|
| 31 |
start_time = time.time()
|
| 32 |
|
|
@@ -78,7 +78,7 @@ with gr.Blocks() as demo:
|
|
| 78 |
realtime = gr.Checkbox(label="Realtime Toggler", info="If TRUE then uses more GPU but create image in realtime.", value=False)
|
| 79 |
latency = gr.Text(label="Latency")
|
| 80 |
with gr.Row():
|
| 81 |
-
seed = gr.Number(label="Seed", value=42
|
| 82 |
randomize_seed = gr.Checkbox(label="Randomize Seed", value=False)
|
| 83 |
with gr.Row():
|
| 84 |
width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_WIDTH)
|
|
|
|
| 26 |
def generate_image(prompt, seed=42, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT, randomize_seed=False, num_inference_steps=2, progress=gr.Progress(track_tqdm=True)):
|
| 27 |
if randomize_seed:
|
| 28 |
seed = random.randint(0, MAX_SEED)
|
| 29 |
+
generator = torch.Generator().manual_seed(int(float(seed)))
|
| 30 |
|
| 31 |
start_time = time.time()
|
| 32 |
|
|
|
|
| 78 |
realtime = gr.Checkbox(label="Realtime Toggler", info="If TRUE then uses more GPU but create image in realtime.", value=False)
|
| 79 |
latency = gr.Text(label="Latency")
|
| 80 |
with gr.Row():
|
| 81 |
+
seed = gr.Number(label="Seed", value=42)
|
| 82 |
randomize_seed = gr.Checkbox(label="Randomize Seed", value=False)
|
| 83 |
with gr.Row():
|
| 84 |
width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_WIDTH)
|