Skip to content

Commit 61d8ad6

Browse files
committed
fix compilation on linux
1 parent ce83041 commit 61d8ad6

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

examples/tts/convert_mimi_to_gguf.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,13 @@
55
from typing import Union
66
from pathlib import Path
77
from torch import Tensor
8-
from transformers import MimiModel
8+
from transformers import MimiModel, PreTrainedModel
99

1010
logger = logging.getLogger("mimi")
1111

1212

1313
class MimiModelConverter:
14-
mimi_model: MimiModel
14+
mimi_model: PreTrainedModel
1515
gguf_writer: gguf.GGUFWriter
1616
fname_out: Path
1717
ftype: gguf.LlamaFileType

examples/tts/mimi-model.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,8 @@
1717
#include <float.h>
1818
#include <cmath>
1919
#include <cstdarg>
20+
#include <functional>
21+
#include <array>
2022

2123
/**
2224
* Implementation of Kyutai's Mimi model using GGML.

0 commit comments

Comments
 (0)