use remote dir

Author: ankitageorge

tests/integration_tests/test_policy_update.py

@@ -251,15 +251,14 @@ async def llama3_torchstore_setup():
 
     store = await MultiProcessStore.create_store()
 
-    model_path = "/tmp/Meta-Llama-3.1-8B-Instruct"
+    model_path = "meta-llama/Meta-Llama-3.1-8B-Instruct"
 
     # Load the model from local path - using device_map="auto" for efficient loading
     model = AutoModelForCausalLM.from_pretrained(
         model_path,
         torch_dtype=torch.float16,  # Use half precision to save memory
         device_map="auto",
         trust_remote_code=True,
-        local_files_only=True,  # Ensure we don't try to download
     )
 
     original_state_dict = model.state_dict()