Spaces:
Build error
Build error
jakewatson commited on
Commit ·
4e2943e
1
Parent(s): 7c5a386
reducing max_tokens
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ import torch
|
|
| 4 |
from transformers import pipeline
|
| 5 |
|
| 6 |
# Inference client setup
|
| 7 |
-
client = InferenceClient("HuggingFaceH4/zephyr-7b-
|
| 8 |
pipe = pipeline("text-generation", "microsoft/Phi-3-mini-4k-instruct", torch_dtype=torch.float32, device_map="auto")
|
| 9 |
|
| 10 |
# Global flag to handle cancellation
|
|
@@ -18,7 +18,7 @@ def respond(
|
|
| 18 |
message,
|
| 19 |
history: list[tuple[str, str]],
|
| 20 |
system_message=base_system_message,
|
| 21 |
-
max_tokens=
|
| 22 |
temperature=0.7,
|
| 23 |
top_p=0.95,
|
| 24 |
use_local_model=False,
|
|
|
|
| 4 |
from transformers import pipeline
|
| 5 |
|
| 6 |
# Inference client setup
|
| 7 |
+
client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
|
| 8 |
pipe = pipeline("text-generation", "microsoft/Phi-3-mini-4k-instruct", torch_dtype=torch.float32, device_map="auto")
|
| 9 |
|
| 10 |
# Global flag to handle cancellation
|
|
|
|
| 18 |
message,
|
| 19 |
history: list[tuple[str, str]],
|
| 20 |
system_message=base_system_message,
|
| 21 |
+
max_tokens=256,
|
| 22 |
temperature=0.7,
|
| 23 |
top_p=0.95,
|
| 24 |
use_local_model=False,
|