gokaygokay
commited on
Commit
•
9576f72
1
Parent(s):
d3e18f0
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 |
|