| 134 | } |
| 135 | |
| 136 | func (m createModel) defaultDiskSizeGB() int { |
| 137 | diskSizeGB := defaultCreateDiskSizeForGPUCount(m.config.NumGPUs) |
| 138 | if m.selectedSnapshot != nil && m.selectedSnapshot.MinimumDiskSizeGB > diskSizeGB { |
| 139 | diskSizeGB = m.selectedSnapshot.MinimumDiskSizeGB |
| 140 | } |
| 141 | if m.specs == nil || m.config.GPUType == "" { |
| 142 | return diskSizeGB |
| 143 | } |
| 144 | |
| 145 | minDisk, maxDisk := m.specs.StorageRange(m.config.GPUType, m.config.NumGPUs) |
| 146 | if m.selectedSnapshot != nil && m.selectedSnapshot.MinimumDiskSizeGB > maxDisk { |
| 147 | maxDisk = m.selectedSnapshot.MinimumDiskSizeGB |
| 148 | } |
| 149 | if diskSizeGB < minDisk { |
| 150 | return minDisk |
| 151 | } |
| 152 | if diskSizeGB > maxDisk { |
| 153 | return maxDisk |
| 154 | } |
| 155 | return diskSizeGB |
| 156 | } |
| 157 | |
| 158 | func (m *createModel) setDefaultDiskSize() { |
| 159 | if m.presets != nil && m.presets.DiskSizeGB != nil { |