[Fix] 🐛 GPU Num Restore Error at Some Scene

This commit is contained in:
Harry-zklcdc
2025-02-16 22:12:28 +08:00
parent bb7b60352e
commit 87995f0572
2 changed files with 2 additions and 1 deletions
@@ -38,6 +38,8 @@ func forceDeleteHandler(ctx iris.Context) {
if instance.FromAction == models.InstanceActionStop || instance.FromAction == models.InstanceActionPause || instance.FromAction == models.InstanceActionRestart {
redis.RawDB.IncrBy(ctx, "remain_gpu:server:"+strconv.Itoa(int(instance.ServerID)), int64(instance.GpuCount))
}
if instance.FromAction != models.InstanceActionCreate {
redis.RawDB.IncrBy(ctx, "remain_volume:server:"+strconv.Itoa(int(instance.ServerID)), int64(instance.VolumeSize+30))
}