Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -151,7 +151,7 @@ def stream(model, code, generate_kwargs):
|
|
151 |
generated_ids = model.generate(**input_ids, **generate_kwargs)
|
152 |
return tokenizer.decode(generated_ids[0][input_ids["input_ids"].shape[1]:], skip_special_tokens=True).strip()
|
153 |
|
154 |
-
@spaces.GPU
|
155 |
def generate(
|
156 |
prompt, temperature=0.6, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, library="LangChain", method="Prefix"
|
157 |
):
|
@@ -170,23 +170,23 @@ def generate(
|
|
170 |
seed=42,
|
171 |
)
|
172 |
|
173 |
-
if method == "Base":
|
174 |
-
|
175 |
-
elif method == "Prefix":
|
176 |
-
|
177 |
-
elif method == "Evo Prefix" and library in ["SQLModel", "SfePy", "MegEngine"]:
|
178 |
-
|
179 |
-
elif method == "FFT" and library in ["SQLModel", "SfePy", "MegEngine"]:
|
180 |
-
|
181 |
-
elif method == "Evo FFT" and library in ["SQLModel", "SfePy", "MegEngine"]:
|
182 |
-
|
183 |
-
elif method == "Full Data FFT" and library in ["SQLModel", "SfePy", "MegEngine"]:
|
184 |
-
|
185 |
-
elif method == "Evo Prefix" and library in ["LangChain", "LlamaIndex", "DSPy"]:
|
186 |
-
|
187 |
-
else:
|
188 |
-
|
189 |
-
|
190 |
return output
|
191 |
|
192 |
|
|
|
151 |
generated_ids = model.generate(**input_ids, **generate_kwargs)
|
152 |
return tokenizer.decode(generated_ids[0][input_ids["input_ids"].shape[1]:], skip_special_tokens=True).strip()
|
153 |
|
154 |
+
@spaces.GPU
|
155 |
def generate(
|
156 |
prompt, temperature=0.6, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0, library="LangChain", method="Prefix"
|
157 |
):
|
|
|
170 |
seed=42,
|
171 |
)
|
172 |
|
173 |
+
# if method == "Base":
|
174 |
+
# output = stream(basemodel, prompt, generate_kwargs)
|
175 |
+
# elif method == "Prefix":
|
176 |
+
# output = stream(model_map[library + " Prefix"], prompt, generate_kwargs)
|
177 |
+
# elif method == "Evo Prefix" and library in ["SQLModel", "SfePy", "MegEngine"]:
|
178 |
+
# output = stream(model_map["Main Evo Prefix"], prompt, generate_kwargs)
|
179 |
+
# elif method == "FFT" and library in ["SQLModel", "SfePy", "MegEngine"]:
|
180 |
+
# output = stream(model_map[library + " FFT"], prompt, generate_kwargs)
|
181 |
+
# elif method == "Evo FFT" and library in ["SQLModel", "SfePy", "MegEngine"]:
|
182 |
+
# output = stream(model_map["Main Evo FFT"], prompt, generate_kwargs)
|
183 |
+
# elif method == "Full Data FFT" and library in ["SQLModel", "SfePy", "MegEngine"]:
|
184 |
+
# output = stream(model_map["Main FD FFT"], prompt, generate_kwargs)
|
185 |
+
# elif method == "Evo Prefix" and library in ["LangChain", "LlamaIndex", "DSPy"]:
|
186 |
+
# output = stream(model_map["CS Evo Prefix"], prompt, generate_kwargs)
|
187 |
+
# else:
|
188 |
+
# output = ""
|
189 |
+
output = ""
|
190 |
return output
|
191 |
|
192 |
|