Skip to content

Commit

Permalink
maintenance commit: nornal is temp==0
Browse files Browse the repository at this point in the history
  • Loading branch information
royerloic committed Dec 28, 2023
1 parent 3b85f0d commit 8b1eecc
Showing 1 changed file with 6 additions and 10 deletions.
16 changes: 6 additions & 10 deletions src/napari_chatgpt/_widget.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,10 +28,10 @@
from arbol import aprint, asection

_creativity_mapping = {}
_creativity_mapping['normal'] = 0.01
_creativity_mapping['slightly creative'] = 0.05
_creativity_mapping['moderately creative'] = 0.1
_creativity_mapping['creative'] = 0.2
_creativity_mapping['normal'] = 0.0
_creativity_mapping['slightly creative'] = 0.01
_creativity_mapping['moderately creative'] = 0.05
_creativity_mapping['creative'] = 0.1


class OmegaQWidget(QWidget):
Expand Down Expand Up @@ -111,21 +111,17 @@ def _model_selection(self):
model_list.append('claude-2')
model_list.append('claude-instant-1')

if is_package_installed('pygpt4all'):
model_list.append('ggml-mpt-7b-chat')
model_list.append('ggml-gpt4all-j-v1.3-groovy')
model_list.append('ggml-gpt4all-l13b-snoozy')

# Postprocess list:
# Ensure that some 'bad' models are at the end of the list:
bad_models = [m for m in model_list if '0613' in m] + ['bard']
bad_models = [m for m in model_list if '0613' in m]
for bad_model in bad_models:
if bad_model in model_list:
model_list.remove(bad_model)
model_list.append(bad_model)

# Ensure that the best models are at the top of the list:
best_models = [m for m in model_list if '0314' in m or '0301' in m]
best_models = [m for m in model_list if '0314' in m or '0301' in m or '1106' in m]
model_list = best_models + [m for m in model_list if m not in best_models]

# normalise list:
Expand Down

0 comments on commit 8b1eecc

Please sign in to comment.