Spaces:
Paused
Paused
Update app.py
Browse files
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 |
|