diff --git a/src/util.go b/src/util.go
index 2b1e32e..9faf2fa 100644
--- a/src/util.go
+++ b/src/util.go
@@ -142,7 +142,7 @@ type Task struct {
 	NumberGPU int    `json:"gpu_number"`
 	MemoryGPU int    `json:"gpu_memory"`
 	IsPS      bool   `json:"is_ps"`
-	ModelGPU  bool   `json:"gpu_model"`
+	ModelGPU  string `json:"gpu_model"`
 }
 
 type Group struct {