Skip to content

Commit 8d9ab9f

Browse files
committed
Upgrade Image Command
- Add image dimension support - Add Upscaling | Resolves #34
1 parent 7149bf0 commit 8d9ab9f

File tree

1 file changed

+25
-2
lines changed

1 file changed

+25
-2
lines changed

interactions/slash/misc/image.js

Lines changed: 25 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,14 +38,14 @@ module.exports = {
3838
.setName("steps")
3939
.setDescription("The Number of Steps to Use")
4040
.setMinValue(1)
41-
.setMaxValue(30),
41+
.setMaxValue(50),
4242
)
4343
.addIntegerOption((option) =>
4444
option
4545
.setName("cfg-scale")
4646
.setDescription("The CFG Scale")
4747
.setMinValue(1)
48-
.setMaxValue(30),
48+
.setMaxValue(20),
4949
)
5050
.addIntegerOption((option) =>
5151
option.setName("seed").setDescription("The Seed").setMinValue(-1),
@@ -108,6 +108,23 @@ module.exports = {
108108
{ name: "PLMS", value: "PLMS" },
109109
{ name: "UniPC", value: "UniPC" },
110110
),
111+
)
112+
.addBooleanOption((o) =>
113+
o.setName("upscale").setDescription("Enable 2x Upscale"),
114+
)
115+
.addIntegerOption((option) =>
116+
option
117+
.setName("width")
118+
.setDescription("The Width of the Image")
119+
.setMinValue(1)
120+
.setMaxValue(1024),
121+
)
122+
.addIntegerOption((option) =>
123+
option
124+
.setName("height")
125+
.setDescription("The Height of the Image")
126+
.setMinValue(1)
127+
.setMaxValue(1024),
111128
),
112129

113130
async execute(interaction) {
@@ -135,6 +152,9 @@ module.exports = {
135152
const cfg_scale = interaction.options.getInteger("cfg-scale");
136153
const seed = interaction.options.getInteger("seed");
137154
const sampler = interaction.options.getString("sampler");
155+
const upscale = interaction.options.getBoolean("upscale");
156+
const width = interaction.options.getInteger("width");
157+
const height = interaction.options.getInteger("height");
138158

139159
let prompt = interaction.options.getString("prompt");
140160
let negative_prompt = interaction.options.getString("negative-prompt");
@@ -231,6 +251,9 @@ module.exports = {
231251
...(sampler && { sampler: sampler }),
232252
...(cfg_scale && { cfg_scale: cfg_scale }),
233253
...(seed && { seed: seed }),
254+
...(upscale && { upscale: upscale }),
255+
...(width && { width: width }),
256+
...(height && { height: height }),
234257
};
235258

236259
try {

0 commit comments

Comments
 (0)