Update app.py
Browse files
app.py
CHANGED
@@ -16,6 +16,7 @@ model = T5ForConditionalGeneration.from_pretrained("roborovski/superprompt-v1",
|
|
16 |
model.to(device)
|
17 |
|
18 |
def generate(your_prompt, task_prefix, max_new_tokens, repetition_penalty, temperature, model_precision_type, top_p, top_k, seed):
|
|
|
19 |
if seed == 0:
|
20 |
seed = random.randint(1, 2**32-1)
|
21 |
transformers.set_seed(seed)
|
@@ -60,7 +61,7 @@ top_p = gr.Slider(value=1, minimum=0, maximum=2, step=0.05, label="Top P", info=
|
|
60 |
|
61 |
top_k = gr.Slider(value=50, minimum=1, maximum=100, step=1, label="Top K", info="Higher k means more diverse outputs by considering a range of tokens")
|
62 |
|
63 |
-
seed = gr.Slider(value=42, minimum=0, maximum=2**32-1, label="Seed", info="A starting point to initiate the generation process, put 0 for a random one")
|
64 |
|
65 |
examples = [
|
66 |
["A storefront with 'Text to Image' written on it.", 512, 1.2, 0.5, "fp16", 1, 50, 42]
|
|
|
16 |
model.to(device)
|
17 |
|
18 |
def generate(your_prompt, task_prefix, max_new_tokens, repetition_penalty, temperature, model_precision_type, top_p, top_k, seed):
|
19 |
+
|
20 |
if seed == 0:
|
21 |
seed = random.randint(1, 2**32-1)
|
22 |
transformers.set_seed(seed)
|
|
|
61 |
|
62 |
top_k = gr.Slider(value=50, minimum=1, maximum=100, step=1, label="Top K", info="Higher k means more diverse outputs by considering a range of tokens")
|
63 |
|
64 |
+
seed = gr.Slider(value=42, minimum=0, maximum=2**32-1, step=1, label="Seed", info="A starting point to initiate the generation process, put 0 for a random one")
|
65 |
|
66 |
examples = [
|
67 |
["A storefront with 'Text to Image' written on it.", 512, 1.2, 0.5, "fp16", 1, 50, 42]
|