MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed / disable

Method disable

deepspeed/linear/optimized_linear.py:120–123  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

118 self.init_lora()
119
120 def disable(self):
121 self.disabled = True
122 self.weight = torch.nn.Parameter(torch.empty((self.output_dim, self.input_dim), dtype=self.dtype),
123 requires_grad=False)
124
125 def init_lora(self):
126 if self.disabled:

Callers 3

_load_from_state_dictMethod · 0.95
disableFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected