Tomtom84 commited on
Commit
29f8312
·
verified ·
1 Parent(s): 55515cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -136,6 +136,7 @@ async def tts(ws: WebSocket):
136
  do_sample=True, temperature=0.7, top_p=0.95,
137
  use_cache=True,
138
  return_dict_in_generate=True,
 
139
  )
140
  print(f"DEBUG: After generate - type of gen.past_key_values: {type(gen.past_key_values)}", flush=True) # Added logging
141
 
 
136
  do_sample=True, temperature=0.7, top_p=0.95,
137
  use_cache=True,
138
  return_dict_in_generate=True,
139
+ cache_position=torch.tensor([offset_len], device=device) # Explicitly pass cache_position
140
  )
141
  print(f"DEBUG: After generate - type of gen.past_key_values: {type(gen.past_key_values)}", flush=True) # Added logging
142