Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 16:17:57 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update the example extension

Browse Source
This commit is contained in:
oobabooga 2023-07-29 18:57:22 -07:00
parent 4b37a2b397
commit ca4188aabc
2 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
docs/Extensions.md
View File

@ -111,14 +111,16 @@ functions are declared in the same order that they are called at
generation time. generation time.
""" """
import gradio as gr
import torch import torch
from modules import chat from transformers import LogitsProcessor
from modules import chat, shared
from modules.text_generation import ( from modules.text_generation import (
decode, decode,
encode, encode,
generate_reply, generate_reply,
) )
from transformers import LogitsProcessor
params = { params = {
"display_name": "Example Extension", "display_name": "Example Extension",

6
extensions/example/script.py
View File

@ -7,14 +7,16 @@ functions are declared in the same order that they are called at
generation time. generation time.
""" """
import gradio as gr
import torch import torch
from modules import chat from transformers import LogitsProcessor
from modules import chat, shared
from modules.text_generation import ( from modules.text_generation import (
decode, decode,
encode, encode,
generate_reply, generate_reply,
) )
from transformers import LogitsProcessor
params = { params = {
"display_name": "Example Extension", "display_name": "Example Extension",