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)
|