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)): for new_name, data_torch in (self.modify_tensors(data_torch, name, bid)):
# TODO: why do we squeeze here? # TODO: why do we squeeze here?
#data = data_torch.squeeze().numpy() # data = data_torch.squeeze().numpy()
data = data_torch.numpy() data = data_torch.numpy()
# if data ends up empty, it means data_torch was a scalar tensor -> restore # if data ends up empty, it means data_torch was a scalar tensor -> restore

View File

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