Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,6 +9,7 @@ import subprocess
|
|
| 9 |
import torch
|
| 10 |
from PIL import Image
|
| 11 |
from transformers import AutoProcessor, AutoModelForCausalLM
|
|
|
|
| 12 |
|
| 13 |
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
| 14 |
|
|
@@ -392,7 +393,7 @@ def create_interface():
|
|
| 392 |
with gr.Row():
|
| 393 |
with gr.Column(scale=2):
|
| 394 |
with gr.Accordion("Basic Settings"):
|
| 395 |
-
seed = gr.Number(label="Seed", value=0)
|
| 396 |
custom = gr.Textbox(label="Custom Input Prompt (optional)")
|
| 397 |
subject = gr.Textbox(label="Subject (optional)")
|
| 398 |
|
|
|
|
| 9 |
import torch
|
| 10 |
from PIL import Image
|
| 11 |
from transformers import AutoProcessor, AutoModelForCausalLM
|
| 12 |
+
import random
|
| 13 |
|
| 14 |
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
| 15 |
|
|
|
|
| 393 |
with gr.Row():
|
| 394 |
with gr.Column(scale=2):
|
| 395 |
with gr.Accordion("Basic Settings"):
|
| 396 |
+
seed = gr.Number(label="Seed", value=random.randint(0, 1000000), randomize=True)
|
| 397 |
custom = gr.Textbox(label="Custom Input Prompt (optional)")
|
| 398 |
subject = gr.Textbox(label="Subject (optional)")
|
| 399 |
|