Added gen3-turbo support

igolaizola · Sep 16, 2024 · 0f6e251 · 0f6e251
1 parent 13be091
commit 0f6e251
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/pkg/cli/cli.go b/pkg/cli/cli.go
@@ -72,7 +72,7 @@ func newGenerateCommand() *ffcli.Command {
 	fs.DurationVar(&cfg.Wait, "wait", 2*time.Second, "wait time between requests")
 	fs.StringVar(&cfg.Token, "token", "", "runway token")
 
-	fs.StringVar(&cfg.Model, "model", "gen2", "model to use (gen2 or gen3)")
+	fs.StringVar(&cfg.Model, "model", "gen3", "model to use (gen2, gen3, gen3-turbo)")
 	fs.StringVar(&cfg.Folder, "folder", "", "runway folder to store assets (optional)")
 	fs.StringVar(&cfg.Image, "image", "", "source image")
 	fs.StringVar(&cfg.Text, "text", "", "source text")

diff --git a/pkg/runway/runway.go b/pkg/runway/runway.go
@@ -488,7 +488,7 @@ func (c *Client) Generate(ctx context.Context, cfg *GenerateRequest) (*Generatio
 			},
 			AsTeamID: c.teamID,
 		}
-	case "gen3":
+	case "gen3", "gen3-turbo":
 		name := fmt.Sprintf("Gen-3 Alpha %d", seed)
 		if len(cfg.Prompt) > 0 {
 			v := cfg.Prompt
@@ -504,8 +504,12 @@ func (c *Client) Generate(ctx context.Context, cfg *GenerateRequest) (*Generatio
 			}
 			name = fmt.Sprintf("%s, %s", name, v)
 		}
+		taskType := "gen3a"
+		if cfg.Model == "gen3-turbo" {
+			taskType = "gen3a_turbo"
+		}
 		createReq = &createGen3TaskRequest{
-			TaskType: "gen3a",
+			TaskType: taskType,
 			Internal: false,
 			Options: gen3Options{
 				Name:            name,