Spaces:
Sleeping
Sleeping
No temp
Browse files
app.py
CHANGED
@@ -27,7 +27,7 @@ def get_completion(prompt, model_name="dolly-v0-70m"):
|
|
27 |
#pipeline_kwargs['torch_dtype'] = "bfloat16"
|
28 |
|
29 |
pipeline_kwargs['max_new_tokens'] = 100
|
30 |
-
pipeline_kwargs['temperature'] = float("inf")
|
31 |
pipeline_kwargs['top_k'] = 1
|
32 |
pipeline_kwargs['top_p'] = 1.0
|
33 |
|
|
|
27 |
#pipeline_kwargs['torch_dtype'] = "bfloat16"
|
28 |
|
29 |
pipeline_kwargs['max_new_tokens'] = 100
|
30 |
+
#pipeline_kwargs['temperature'] = float("inf")
|
31 |
pipeline_kwargs['top_k'] = 1
|
32 |
pipeline_kwargs['top_p'] = 1.0
|
33 |
|