diff --git a/modeling/inference_models/hf_torch.py b/modeling/inference_models/hf_torch.py index 5dd53bf8..47c37436 100644 --- a/modeling/inference_models/hf_torch.py +++ b/modeling/inference_models/hf_torch.py @@ -820,7 +820,7 @@ class HFTorchInferenceModel(HFInferenceModel): breakmodel.gpu_blocks = [0] * n_layers return - elif breakmodel.gpu_blocks != []: + elif breakmodel.gpu_blocks == []: logger.info("Breakmodel not specified, assuming GPU 0") breakmodel.gpu_blocks = [n_layers] n_layers = 0