debisoft commited on
Commit
87323b0
·
1 Parent(s): 49e2dd3
Files changed (1) hide show
  1. app.py +1 -1
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