chore: update .env.example with device setting

atrifat · Jul 19, 2024 · ad4adaf · ad4adaf
1 parent b5070f8
commit ad4adaf
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/.env.example b/.env.example
@@ -15,4 +15,7 @@ TOPIC_CLASSIFICATION_MODEL="cardiffnlp/twitter-roberta-base-dec2021-tweet-topic-
 ENABLE_CACHE=false
 
 # (Optional. Default: 60. Options: integer value) Duration of cache in seconds. Parameter will be used when the cache is enabled
-CACHE_DURATION_SECONDS=60
+CACHE_DURATION_SECONDS=60
+
+# (Optional. Default: auto. Options: auto,cpu,cuda) Set torch default device for pipeline. Automatically detect if cuda/gpu device is available
+TORCH_DEVICE=auto