made changes to readme
Browse files
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'
|
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.
|