Skip to content

Commit

Permalink
Fix problem reported by lint
Browse files Browse the repository at this point in the history
  • Loading branch information
mikecovlee committed Aug 30, 2023
1 parent 05b5cf2 commit 55092b8
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion aspen/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ def forward(self, data: torch.Tensor) -> torch.Tensor:

class Linear():
def __init__(self, weight: torch.Tensor, load_in_8bit: bool = True, device: str = None):
if device == None:
if device is None:
device = weight.device
row, col = weight.shape
if load_in_8bit:
Expand Down
2 changes: 1 addition & 1 deletion mlora.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ def log(msg: str):
exit(-1)


if args.model_name_or_path == None:
if args.model_name_or_path is None:
print('error: Argument --model_name_or_path are required.')
parser.print_help()
exit(-1)
Expand Down

0 comments on commit 55092b8

Please sign in to comment.