critical change

This commit is contained in:
somebody
2023-07-24 17:17:57 -05:00
parent a73420c49c
commit ad4528b5a6

View File

@@ -103,7 +103,6 @@ def load_quant_offload_device_map(
print(device_map) print(device_map)
m, layers, remaining = find_layers(model) m, layers, remaining = find_layers(model)
type(m).non_offload_forward = type(m).forward type(m).non_offload_forward = type(m).forward
# Hook offload_forward into found model # Hook offload_forward into found model