Ver Fonte

Merge branch 'main' into state_as_function_params

oobabooga há 2 anos atrás
pai
commit
cf239c1232
2 ficheiros alterados com 7 adições e 0 exclusões
  1. 1 0
      modules/shared.py
  2. 6 0
      presets/LLaMA-Precise.txt

+ 1 - 0
modules/shared.py

@@ -45,6 +45,7 @@ settings = {
     'chat_default_extensions': ["gallery"],
     'presets': {
         'default': 'NovelAI-Sphinx Moth',
+        '.*(alpaca|llama)': "LLaMA-Precise",
         '.*pygmalion': 'NovelAI-Storywriter',
         '.*RWKV': 'Naive',
     },

+ 6 - 0
presets/LLaMA-Precise.txt

@@ -0,0 +1,6 @@
+do_sample=True
+top_p=0.1
+top_k=40
+temperature=0.7
+repetition_penalty=1.18
+typical_p=1.0