Quellcode durchsuchen

Add seed to settings

oobabooga vor 2 Jahren
Ursprung
Commit
1d1d9e40cd
3 geänderte Dateien mit 3 neuen und 1 gelöschten Zeilen
  1. 1 0
      modules/shared.py
  2. 1 1
      server.py
  3. 1 0
      settings-template.json

+ 1 - 0
modules/shared.py

@@ -27,6 +27,7 @@ settings = {
     'max_new_tokens': 200,
     'max_new_tokens_min': 1,
     'max_new_tokens_max': 2000,
+    'seed': -1,
     'name1': 'You',
     'name2': 'Assistant',
     'context': 'This is a conversation with your Assistant. The Assistant is very helpful and is eager to chat with you and answer your questions.',

+ 1 - 1
server.py

@@ -166,7 +166,7 @@ def create_settings_menus(default_preset):
         with gr.Column():
             create_model_and_preset_menus()
         with gr.Column():
-            shared.gradio['seed'] = gr.Number(value=-1, label='Seed (-1 for random)')
+            shared.gradio['seed'] = gr.Number(value=shared.settings['seed'], label='Seed (-1 for random)')
 
     with gr.Row():
         with gr.Column():

+ 1 - 0
settings-template.json

@@ -2,6 +2,7 @@
     "max_new_tokens": 200,
     "max_new_tokens_min": 1,
     "max_new_tokens_max": 2000,
+    "seed": -1,
     "name1": "You",
     "name2": "Assistant",
     "context": "This is a conversation with your Assistant. The Assistant is very helpful and is eager to chat with you and answer your questions.",