jakewatson commited on
Commit
4e2943e
·
1 Parent(s): 7c5a386

reducing max_tokens

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-alpha")
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=512,
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,