IST199655
commited on
Commit
·
d44f7f3
1
Parent(s):
b6e525e
no cuda
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ def respond(
|
|
51 |
tokenize = True,
|
52 |
add_generation_prompt = True, # Must add for generation
|
53 |
return_tensors = "pt",
|
54 |
-
)
|
55 |
|
56 |
outputs = model.generate(input_ids = inputs, max_new_tokens = max_tokens, use_cache = True,
|
57 |
temperature = 1.5, min_p = 0.1)
|
|
|
51 |
tokenize = True,
|
52 |
add_generation_prompt = True, # Must add for generation
|
53 |
return_tensors = "pt",
|
54 |
+
)
|
55 |
|
56 |
outputs = model.generate(input_ids = inputs, max_new_tokens = max_tokens, use_cache = True,
|
57 |
temperature = 1.5, min_p = 0.1)
|