aisingapore
/

sea-lion-7b-instruct-gptq

Text Generation

text-generation-inference

Model card Files Files and versions Community

sea-lion-7b-instruct-gptq / fc.py

RaymondAISG's picture

Upload sea-lion-7b-gptq

c6f2274 verified 3 months ago

raw history blame contribute delete

No virus

170 Bytes

	from torch import nn

	FC_CLASS_REGISTRY = {"torch": nn.Linear}
	try:
	import transformer_engine.pytorch as te

	FC_CLASS_REGISTRY["te"] = te.Linear
	except:
	pass