Update config.py

config.py

@@ -1,8 +1,8 @@
 from dataclasses import dataclass
 from typing import Optional
-
+import os
 import torch
-
+from dotenv import load_dotenv
 from cpufeature import CPUFeature
 from petals.constants import PUBLIC_INITIAL_PEERS
 
@@ -12,7 +12,9 @@ class ModelInfo:
     repo: str
     adapter: Optional[str] = None
 
-
+load_dotenv()
+hugging_face_token = os.getenv("HUGGINGFACE_TOKEN")
+login(token=hugging_face_token)
 MODELS = [
     ModelInfo(repo="meta-llama/Llama-2-70b-hf"),
     ModelInfo(repo="meta-llama/Llama-2-70b-chat-hf"),