We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ce83041 commit 61d8ad6Copy full SHA for 61d8ad6
examples/tts/convert_mimi_to_gguf.py
@@ -5,13 +5,13 @@
5
from typing import Union
6
from pathlib import Path
7
from torch import Tensor
8
-from transformers import MimiModel
+from transformers import MimiModel, PreTrainedModel
9
10
logger = logging.getLogger("mimi")
11
12
13
class MimiModelConverter:
14
- mimi_model: MimiModel
+ mimi_model: PreTrainedModel
15
gguf_writer: gguf.GGUFWriter
16
fname_out: Path
17
ftype: gguf.LlamaFileType
examples/tts/mimi-model.cpp
@@ -17,6 +17,8 @@
#include <float.h>
18
#include <cmath>
19
#include <cstdarg>
20
+#include <functional>
21
+#include <array>
22
23
/**
24
* Implementation of Kyutai's Mimi model using GGML.
0 commit comments