PB Unity commited on
Commit
55d35e7
1 Parent(s): f931e60

Upload RunMusicGen.cs

Browse files
Files changed (1) hide show
  1. 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(BackendType.GPUCompute, null);
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(BackendType.GPUCompute, decoder);
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()