IST199655 commited on
Commit
d44f7f3
·
1 Parent(s): b6e525e
Files changed (1) hide show
  1. app.py +1 -1
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
- ).to("cuda")
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)