Credit turboderp

This commit is contained in:
oobabooga 2023-08-06 13:42:43 -07:00
parent 0af10ab49b
commit d4b851bdc8

View File

@ -136,6 +136,7 @@ class ExllamaModel:
break
# Case 2: CFG
# Copied from https://github.com/turboderp/exllama/blob/master/example_cfg.py
else:
alpha = state['guidance_scale']
prompts = [prompt, state['negative_prompt'] or '']