Update app.py
Browse files
app.py
CHANGED
@@ -30,7 +30,7 @@ footer {
|
|
30 |
# Initialize the tokenizer
|
31 |
tokenizer = AutoTokenizer.from_pretrained(model)
|
32 |
|
33 |
-
|
34 |
|
35 |
@spaces.GPU
|
36 |
def generate(message, history, system, max_tokens, temperature, top_p, top_k, penalty):
|
@@ -58,6 +58,7 @@ def generate(message, history, system, max_tokens, temperature, top_p, top_k, pe
|
|
58 |
eos_token_id=[151645,151643],
|
59 |
)
|
60 |
# generate outputs
|
|
|
61 |
outputs = llm.generate([text], sampling_params)
|
62 |
|
63 |
# Print the outputs.
|
|
|
30 |
# Initialize the tokenizer
|
31 |
tokenizer = AutoTokenizer.from_pretrained(model)
|
32 |
|
33 |
+
|
34 |
|
35 |
@spaces.GPU
|
36 |
def generate(message, history, system, max_tokens, temperature, top_p, top_k, penalty):
|
|
|
58 |
eos_token_id=[151645,151643],
|
59 |
)
|
60 |
# generate outputs
|
61 |
+
llm = LLM(model=model)
|
62 |
outputs = llm.generate([text], sampling_params)
|
63 |
|
64 |
# Print the outputs.
|