Function GET_DTYPE

lightx2v/utils/envs.py:32–35 · view source on GitHub ↗

()

Source from the content-addressed store, hash-verified

30
31	@lru_cache(maxsize=None)
32	def GET_DTYPE():
33	RUNNING_FLAG = os.getenv("DTYPE", "BF16")
34	assert RUNNING_FLAG in ["BF16", "FP16"]
35	return DTYPE_MAP[RUNNING_FLAG]
36
37
38	@lru_cache(maxsize=None)

send_encoder_outputsMethod · 0.90

receive_encoder_outputsMethod · 0.90

send_transformer_outputsMethod · 0.90

receive_transformer_outputsMethod · 0.90

load_wan_vae_encoderFunction · 0.90

load_wan_vae_decoderFunction · 0.90

processMethod · 0.90

_get_vae_encoder_outputMethod · 0.90

processMethod · 0.90

get_vae_encoder_outputFunction · 0.90

mainFunction · 0.90

no outgoing calls

no test coverage detected