PB Unity
commited on
Upload RunMusicGen.cs
Browse files- RunMusicGen.cs +4 -2
RunMusicGen.cs
CHANGED
@@ -36,6 +36,8 @@ public class RunMusicGen : MonoBehaviour
|
|
36 |
// Make this value smaller to make music more random
|
37 |
float predictability = 1f;
|
38 |
|
|
|
|
|
39 |
public AudioClip clip;
|
40 |
IWorker toWavEngine, decoderEngine, textEngine, projectEngine;
|
41 |
|
@@ -72,7 +74,7 @@ public class RunMusicGen : MonoBehaviour
|
|
72 |
bool hasDecodedMusic = false;
|
73 |
void Start()
|
74 |
{
|
75 |
-
ops = WorkerFactory.CreateOps(
|
76 |
|
77 |
maxFrames = 50 * seconds + 3;
|
78 |
|
@@ -94,7 +96,7 @@ public class RunMusicGen : MonoBehaviour
|
|
94 |
void LoadDecoderModel()
|
95 |
{
|
96 |
decoder = ModelLoader.Load(Application.streamingAssetsPath + "/decoder.sentis");
|
97 |
-
decoderEngine = WorkerFactory.CreateWorker(
|
98 |
}
|
99 |
|
100 |
void CreateAttentionMask()
|
|
|
36 |
// Make this value smaller to make music more random
|
37 |
float predictability = 1f;
|
38 |
|
39 |
+
BackendType backendType = BackendType.GPUCompute;
|
40 |
+
|
41 |
public AudioClip clip;
|
42 |
IWorker toWavEngine, decoderEngine, textEngine, projectEngine;
|
43 |
|
|
|
74 |
bool hasDecodedMusic = false;
|
75 |
void Start()
|
76 |
{
|
77 |
+
ops = WorkerFactory.CreateOps(backendType, null);
|
78 |
|
79 |
maxFrames = 50 * seconds + 3;
|
80 |
|
|
|
96 |
void LoadDecoderModel()
|
97 |
{
|
98 |
decoder = ModelLoader.Load(Application.streamingAssetsPath + "/decoder.sentis");
|
99 |
+
decoderEngine = WorkerFactory.CreateWorker(backendType, decoder);
|
100 |
}
|
101 |
|
102 |
void CreateAttentionMask()
|