Skip to content

Commit b8026ba

Browse files
committed
chore: update .env.example with torch setting
1 parent 0b9f81f commit b8026ba

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

.env.example

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,4 +10,7 @@ DETOXIFY_MODEL=unbiased-small
1010
ENABLE_CACHE=false
1111

1212
# (Optional. Default: 60. Options: integer value) Duration of cache in seconds. Parameter will be used when the cache is enabled
13-
CACHE_DURATION_SECONDS=60
13+
CACHE_DURATION_SECONDS=60
14+
15+
# (Optional. Default: auto. Options: auto,cpu,cuda) Set torch default device for detoxify library. Automatically detect if cuda/gpu device is available
16+
TORCH_DEVICE=auto

0 commit comments

Comments
 (0)