We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9b420cc commit a90003fCopy full SHA for a90003f
src/peft/tuners/_buffer_dict.py
@@ -148,7 +148,8 @@ def extra_repr(self):
148
child_lines = []
149
for k, p in self._buffers.items():
150
size_str = "x".join(str(size) for size in p.size())
151
- device_str = "" if not p.is_cuda else f" (GPU {p.get_device()})"
+ device_type = p.device.type
152
+ device_str = "" if device_type == "cpu" else f" ({device_type.upper()} {p.get_device()})"
153
parastr = f"Buffer containing: [{torch.typename(p)} of size {size_str}{device_str}]"
154
child_lines.append(" (" + k + "): " + parastr)
155
tmpstr = "\n".join(child_lines)
0 commit comments