Commit 74c00eb1 authored by Ruslan Svirschevski's avatar Ruslan Svirschevski
Browse files

remove moving offset if not nested

parent 070f45d2
......@@ -680,7 +680,6 @@ class QuantState:
def to(self, device):
# make sure the quantization state is on the right device
self.absmax = self.absmax.to(device)
self.offset = self.offset.to(device)
if self.nested:
self.offset = self.offset.to(device)
self.state2.absmax = self.state2.absmax.to(device)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment