Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -72,7 +72,6 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
|
|
72 |
print(datetime.datetime.now())
|
73 |
uid=uuid.uuid4()
|
74 |
|
75 |
-
seed = random.randint(1,1111111111111111)
|
76 |
|
77 |
agent=prompts.AI_REPORT_WRITER
|
78 |
''' if agent_name == "WEB_DEV":
|
@@ -97,15 +96,18 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
|
|
97 |
temperature = 1e-2
|
98 |
top_p = float(top_p)
|
99 |
|
100 |
-
|
101 |
-
temperature=temperature,
|
102 |
-
max_new_tokens=1048,
|
103 |
-
top_p=top_p,
|
104 |
-
repetition_penalty=repetition_penalty,
|
105 |
-
do_sample=True,
|
106 |
-
seed=seed,
|
107 |
-
)
|
108 |
while True:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
109 |
formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
|
110 |
stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
|
111 |
output = ""
|
|
|
72 |
print(datetime.datetime.now())
|
73 |
uid=uuid.uuid4()
|
74 |
|
|
|
75 |
|
76 |
agent=prompts.AI_REPORT_WRITER
|
77 |
''' if agent_name == "WEB_DEV":
|
|
|
96 |
temperature = 1e-2
|
97 |
top_p = float(top_p)
|
98 |
|
99 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
100 |
while True:
|
101 |
+
seed = random.randint(1,1111111111111111)
|
102 |
+
|
103 |
+
generate_kwargs = dict(
|
104 |
+
temperature=temperature,
|
105 |
+
max_new_tokens=1048,
|
106 |
+
top_p=top_p,
|
107 |
+
repetition_penalty=repetition_penalty,
|
108 |
+
do_sample=True,
|
109 |
+
seed=seed,
|
110 |
+
)
|
111 |
formatted_prompt = format_prompt(f"{system_prompt}, {prompt}", history)
|
112 |
stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
|
113 |
output = ""
|