diff --git a/cmd/model.go b/cmd/model.go
index 34453a1..8ca0452 100644
--- a/cmd/model.go
+++ b/cmd/model.go
@@ -51,7 +51,10 @@ var modelCmd = &cobra.Command{
 					log.Fatal(err)
 				}
 			case database.DriverSQLite:
-
+				err := gq.DB.Raw("SELECT name FROM sqlite_master WHERE type='table'").Scan(&tables).Error
+				if err != nil {
+					log.Fatal(err)
+				}
 			}
 
 			if len(tables) == 0 {
diff --git a/config.toml b/config.toml
index e597f03..7351a1e 100644
--- a/config.toml
+++ b/config.toml
@@ -88,7 +88,7 @@ ShowLine = true
 LogInConsole = true
 
 [Database]
-Driver = "postgres"
+Driver = "sqlite"
 
 [Database.MySQL]
 Host = "10.47.119.226"
@@ -116,7 +116,7 @@ MaxIdleConns = 10
 MaxOpenConns = 200
 
 [Database.SQLite]
-File = "sqlite.db"
+File = "data.db"
 
 [Database.Redis]
 Host = "10.47.119.226"