Merge branch 'main' into nightly
This commit is contained in:
commit
074aa737ce
1 changed files with 1 additions and 3 deletions
|
|
@ -1173,7 +1173,7 @@ class FastLlamaModel:
|
|||
# Downcast RoPE embedding to correct data type
|
||||
if (name.endswith("rotary_emb") or hasattr(module, "cos_cached")) \
|
||||
and (module.cos_cached.dtype != correct_dtype):
|
||||
|
||||
|
||||
module.cos_cached = module.cos_cached.to(correct_dtype)
|
||||
module.sin_cached = module.sin_cached.to(correct_dtype)
|
||||
pass
|
||||
|
|
@ -1216,8 +1216,6 @@ class FastLlamaModel:
|
|||
)
|
||||
pass
|
||||
|
||||
if loftq_config is None: loftq_config = {}
|
||||
|
||||
import inspect
|
||||
signature = str(inspect.signature(LoraConfig))
|
||||
SUPPORTS_LOFTQ = "loftq_config" in signature
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue