Update orpheus-tts/engine_class.py
Browse files
orpheus-tts/engine_class.py
CHANGED
|
@@ -101,7 +101,6 @@ class OrpheusModel:
|
|
| 101 |
full_prompt = prompt
|
| 102 |
|
| 103 |
# Kartoffel model format - exakt wie in der Referenz-Implementierung
|
| 104 |
-
import torch
|
| 105 |
|
| 106 |
start_token = torch.tensor([[128259]], dtype=torch.int64)
|
| 107 |
end_tokens = torch.tensor([[128009, 128260]], dtype=torch.int64)
|
|
|
|
| 101 |
full_prompt = prompt
|
| 102 |
|
| 103 |
# Kartoffel model format - exakt wie in der Referenz-Implementierung
|
|
|
|
| 104 |
|
| 105 |
start_token = torch.tensor([[128259]], dtype=torch.int64)
|
| 106 |
end_tokens = torch.tensor([[128009, 128260]], dtype=torch.int64)
|