o پi @s8ddlZddlmZddlZeeZGdddZdS)N)Optionalc@s@eZdZUeejed<eeed<d dededdfddZ dS) DeviceConfigdevicegpu_idcudareturnNcCs6|dvr||_ntd|t|j|_||_dS)N)rxpuhpucpunpumusazNot supported device type: ) device_type RuntimeErrortorchrr)selfrrrT/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/configs/device_config.py__init__ s  zDeviceConfig.__init__)rr) __name__ __module__ __qualname__rrr__annotations__intstrrrrrrr s  r)loggingtypingrr getLoggerrloggerrrrrrs