NousResearch
/

OLMo-Bitnet-1B

Text Generation

Model card Files Files and versions

emozilla commited on Mar 30, 2024

Commit

2e055bf

·

1 Parent(s): 2b1c7b3

update inference code

Files changed (1) hide show

optim.py +1 -1

optim.py CHANGED Viewed

@@ -10,7 +10,7 @@ import torch.nn as nn
 from torch.distributed.fsdp import FullyShardedDataParallel
 from torch.optim.optimizer import Optimizer as OptimizerBase
-from . import LayerNormBase, BitLinear158
 from .config import OptimizerType, SchedulerConfig, SchedulerType, TrainConfig
 from .torch_util import get_default_device, is_distributed

 from torch.distributed.fsdp import FullyShardedDataParallel
 from torch.optim.optimizer import Optimizer as OptimizerBase
+from .model import LayerNormBase, BitLinear158
 from .config import OptimizerType, SchedulerConfig, SchedulerType, TrainConfig
 from .torch_util import get_default_device, is_distributed