tts : add header + minor fixes

ggml-ci
This commit is contained in:
Georgi Gerganov 2024-12-11 19:00:03 +02:00
parent eabc921355
commit a92f4137fb
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
2 changed files with 4 additions and 3 deletions

View File

@ -297,7 +297,7 @@ class Model:
for new_name, data_torch in (self.modify_tensors(data_torch, name, bid)):
# TODO: why do we squeeze here?
#data = data_torch.squeeze().numpy()
# data = data_torch.squeeze().numpy()
data = data_torch.numpy()
# if data ends up empty, it means data_torch was a scalar tensor -> restore

View File

@ -6,6 +6,7 @@
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <string>
#include <vector>
#include <fstream>
@ -354,9 +355,9 @@ int main(int argc, char ** argv) {
// TODO: not sure if this is correct
{
std::string prompt_clean;
std::string prompt_user = params.prompt;
std::string prompt_clean = "";
//string_replace_all(prompt_user, " ", "<|text_sep|>");
for (auto & c : prompt_user) {
if (c == ' ') {
prompt_clean += "<|text_sep|>";