bad

higgsfield-ai · Oct 9, 2023 · f5f06de · f5f06de
1 parent cdf947f
commit f5f06de
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/internal/docker.go b/internal/docker.go
@@ -33,6 +33,7 @@ const (
 	imageTag       = "hf-torch:latest"
 	guestRootPath  = "/srv/"
 	guestCachePath = "/home/nonroot/.cache/"
+  guestRootCachePath = "/root/.cache/"
 )
 
 func NewDockerRun(
@@ -166,6 +167,7 @@ func (d *DockerRun) Run(
 			Binds: []string{
 				fmt.Sprintf("%s:%s", d.hostRootPath, d.guestRootPath),
 				fmt.Sprintf("%s:%s", d.hostCachePath, d.guestCachePath),
+        fmt.Sprintf("%s:%s", d.hostCachePath, guestRootCachePath),
 			},
 			IpcMode:     container.IPCModeHost,
 			PidMode:     container.PidMode("host"),