pmolchanov commited on
Commit
ff5a6e0
·
verified ·
1 Parent(s): ae2b257

Update app_chat.py

Browse files
Files changed (1) hide show
  1. app_chat.py +6 -2
app_chat.py CHANGED
@@ -9,8 +9,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStream
9
  from transformers import StoppingCriteria, StoppingCriteriaList, StopStringCriteria
10
 
11
  import subprocess
12
- # useradd -m -u 1000 user
13
- subprocess.run(['useradd', '-m', "-u", "1000", "user" ])
 
14
  import torch._dynamo
15
  torch._dynamo.config.suppress_errors = True
16
 
@@ -57,6 +58,9 @@ def generate(
57
  repetition_penalty: float = 1.2,
58
  ) -> Iterator[str]:
59
  conversation = []
 
 
 
60
  if system_prompt:
61
  conversation.append({"role": "system", "content": system_prompt})
62
  conversation += chat_history
 
9
  from transformers import StoppingCriteria, StoppingCriteriaList, StopStringCriteria
10
 
11
  import subprocess
12
+
13
+ global model
14
+
15
  import torch._dynamo
16
  torch._dynamo.config.suppress_errors = True
17
 
 
58
  repetition_penalty: float = 1.2,
59
  ) -> Iterator[str]:
60
  conversation = []
61
+
62
+ global model
63
+
64
  if system_prompt:
65
  conversation.append({"role": "system", "content": system_prompt})
66
  conversation += chat_history