(&self, name: &str)
| 270 | } |
| 271 | |
| 272 | fn contains_tensor(&self, name: &str) -> bool { |
| 273 | // A tensor is "present" if either its direct name, GPTQ qweight, or scales exist. |
| 274 | if self.inner.get(name).is_ok() { |
| 275 | return true; |
| 276 | } |
| 277 | let prefix = name.strip_suffix(".weight").unwrap_or(name); |
| 278 | self.inner.get(&format!("{prefix}.qweight")).is_ok() |
| 279 | || self.inner.get(&format!("{prefix}.scales")).is_ok() |
| 280 | } |
| 281 | } |
| 282 | |
| 283 | /// Create a VarBuilder that transparently dequantizes GPTQ 4-bit weights. |