Fix loras.

This commit is contained in:
comfyanonymous 2025-12-05 02:05:14 -05:00 committed by GitHub
parent e12842fe31
commit 129f3e7db1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -338,7 +338,9 @@ def generic_copy_(func, args, kwargs):
# Copy from another quantized tensor # Copy from another quantized tensor
qt_dest._qdata.copy_(src._qdata, non_blocking=non_blocking) qt_dest._qdata.copy_(src._qdata, non_blocking=non_blocking)
qt_dest._layout_type = src._layout_type qt_dest._layout_type = src._layout_type
orig_dtype = qt_dest._layout_params["orig_dtype"]
_copy_layout_params_inplace(src._layout_params, qt_dest._layout_params, non_blocking=non_blocking) _copy_layout_params_inplace(src._layout_params, qt_dest._layout_params, non_blocking=non_blocking)
qt_dest._layout_params["orig_dtype"] = orig_dtype
else: else:
# Copy from regular tensor - just copy raw data # Copy from regular tensor - just copy raw data
qt_dest._qdata.copy_(src) qt_dest._qdata.copy_(src)