mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-28 01:47:32 +08:00
Update gpt_oss.py
This commit is contained in:
parent
874e5e7890
commit
6326a91711
@ -608,7 +608,7 @@ def lens_te(dtype_llama=None, llama_quantization_metadata=None):
|
||||
mo = dict(model_options or {})
|
||||
if llama_quantization_metadata is not None:
|
||||
mo["quantization_metadata"] = llama_quantization_metadata
|
||||
if dtype_llama is not None:
|
||||
if dtype is None and dtype_llama is not None:
|
||||
dtype = dtype_llama
|
||||
super().__init__(device=device, dtype=dtype, model_options=mo)
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user