wanzin commited on
Commit
d0f5011
1 Parent(s): 9f93075

made changes to readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -35,7 +35,7 @@ description: >-
35
  - **batch_size** (`Optional[Union[int, str]]`): Batch size for evaluation, defaults to None.
36
  - **max_batch_size** (`Optional[int]`): Maximum batch size to try with automatic batch size detection, defaults to None.
37
  - **limit** (`Optional[Union[int, float]]`): Limit the number of examples per task, defaults to None.
38
- - **device** (`Optional[str]`): Device to run on, defaults to 'cuda' when available, otherwise 'cpu'.
39
  - **revision** (`str`): Model revision to use, defaults to 'main'.
40
  - **trust_remote_code** (`bool`): Whether to trust remote code, defaults to False.
41
  - **log_samples** (`bool`): If True, logs all model outputs and documents, defaults to True.
 
35
  - **batch_size** (`Optional[Union[int, str]]`): Batch size for evaluation, defaults to None.
36
  - **max_batch_size** (`Optional[int]`): Maximum batch size to try with automatic batch size detection, defaults to None.
37
  - **limit** (`Optional[Union[int, float]]`): Limit the number of examples per task, defaults to None.
38
+ - **device** (`Optional[str]`): Device to run on. If None, defaults to 'cuda' if available, otherwise 'cpu'.
39
  - **revision** (`str`): Model revision to use, defaults to 'main'.
40
  - **trust_remote_code** (`bool`): Whether to trust remote code, defaults to False.
41
  - **log_samples** (`bool`): If True, logs all model outputs and documents, defaults to True.