Skip to content

Commit 115f25a

Browse files
Move sampler classes to dedicated package
1 parent a2ec8cc commit 115f25a

File tree

7 files changed

+9
-9
lines changed

7 files changed

+9
-9
lines changed

src/main/java/com/example/LlamaApp.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,9 @@
22

33
import com.example.aot.AOT;
44
import com.example.core.model.tensor.FloatTensor;
5-
import com.example.inference.CategoricalSampler;
6-
import com.example.inference.Sampler;
7-
import com.example.inference.ToppSampler;
5+
import com.example.inference.sampler.CategoricalSampler;
6+
import com.example.inference.sampler.Sampler;
7+
import com.example.inference.sampler.ToppSampler;
88
import com.example.model.Model;
99
import com.example.loader.weights.ModelLoader;
1010
import com.example.tornadovm.FloatArrayUtils;

src/main/java/com/example/inference/CategoricalSampler.java renamed to src/main/java/com/example/inference/sampler/CategoricalSampler.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package com.example.inference;
1+
package com.example.inference.sampler;
22

33
import com.example.core.model.tensor.FloatTensor;
44
import uk.ac.manchester.tornado.api.types.arrays.FloatArray;

src/main/java/com/example/inference/Sampler.java renamed to src/main/java/com/example/inference/sampler/Sampler.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package com.example.inference;
1+
package com.example.inference.sampler;
22

33
import com.example.core.model.tensor.FloatTensor;
44
import uk.ac.manchester.tornado.api.types.arrays.FloatArray;

src/main/java/com/example/inference/ToppSampler.java renamed to src/main/java/com/example/inference/sampler/ToppSampler.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
package com.example.inference;
1+
package com.example.inference.sampler;
22

33
import com.example.core.model.tensor.FloatTensor;
44
import uk.ac.manchester.tornado.api.types.arrays.FloatArray;

src/main/java/com/example/model/Model.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
package com.example.model;
22

3-
import com.example.inference.Sampler;
3+
import com.example.inference.sampler.Sampler;
44
import com.example.Options;
55
import com.example.loader.weights.ModelLoader.ModelType;
66
import com.example.loader.weights.State;

src/main/java/com/example/model/llama/Llama.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
import com.example.auxiliary.Parallel;
44
import com.example.auxiliary.format.LlamaChatFormat;
55
import com.example.core.model.tensor.FloatTensor;
6-
import com.example.inference.Sampler;
76
import com.example.model.Configuration;
7+
import com.example.inference.sampler.Sampler;
88
import com.example.model.Model;
99
import com.example.Options;
1010
import com.example.loader.weights.ModelLoader;

src/main/java/com/example/model/mistral/Mistral.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
import com.example.auxiliary.Parallel;
44
import com.example.auxiliary.format.MistralChatFormat;
55
import com.example.core.model.tensor.FloatTensor;
6-
import com.example.inference.Sampler;
76
import com.example.model.Configuration;
7+
import com.example.inference.sampler.Sampler;
88
import com.example.model.Model;
99
import com.example.Options;
1010
import com.example.loader.weights.ModelLoader;

0 commit comments

Comments
 (0)