Merge pull request #42 from SmilyOrg/disable-ai-load

Disable loading AI if not available
SmilyOrg · Oct 11, 2022 · 71ab607 · 71ab607
2 parents 747e9b9 + 126f303
commit 71ab607
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/internal/image/source.go b/internal/image/source.go
@@ -133,14 +133,15 @@ func NewSource(config Config, migrations embed.FS) *Source {
 		go source.ColorQueue.Run()
 
 		source.Clip = config.AI
-
-		source.AIQueue = queue.Queue{
-			ID:          "load_ai",
-			Name:        "load ai",
-			Worker:      source.loadInfosAI,
-			WorkerCount: 8,
+		if config.AI.Available() {
+			source.AIQueue = queue.Queue{
+				ID:          "load_ai",
+				Name:        "load ai",
+				Worker:      source.loadInfosAI,
+				WorkerCount: 8,
+			}
+			go source.AIQueue.Run()
 		}
-		go source.AIQueue.Run()
 	}
 
 	return &source