Update inference_fine_tune.py
Browse files- inference_fine_tune.py +2 -2
inference_fine_tune.py
CHANGED
|
@@ -43,8 +43,8 @@ def generate_response(prompt: str, history):
|
|
| 43 |
return
|
| 44 |
|
| 45 |
input_tokens = torch.tensor(input_tokens).unsqueeze(0).to(device)
|
| 46 |
-
temperature = 0.
|
| 47 |
-
top_k =
|
| 48 |
generated_text = ""
|
| 49 |
i = 0
|
| 50 |
|
|
|
|
| 43 |
return
|
| 44 |
|
| 45 |
input_tokens = torch.tensor(input_tokens).unsqueeze(0).to(device)
|
| 46 |
+
temperature = 0.5
|
| 47 |
+
top_k = 20
|
| 48 |
generated_text = ""
|
| 49 |
i = 0
|
| 50 |
|