mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-06-05 21:59:24 +02:00
critical change
This commit is contained in:
@@ -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
|
||||||
|
Reference in New Issue
Block a user