Update test.txt
Browse files
test.txt
CHANGED
|
@@ -1,14 +1,3 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
Write-Host "--- Disk space on current drive ---"
|
| 5 |
-
Get-PSDrive -Name (Get-Location).Drive.Name
|
| 6 |
-
|
| 7 |
-
Write-Host "--- Package versions ---"
|
| 8 |
-
.\venv\Scripts\python.exe -c "import torch, transformers, safetensors, accelerate; print('torch', torch.__version__); print('transformers', transformers.__version__); print('safetensors', safetensors.__version__); print('accelerate', accelerate.__version__)"
|
| 9 |
-
|
| 10 |
-
Write-Host "--- Isolated load test ---"
|
| 11 |
-
.\venv\Scripts\python.exe -c "from transformers import AutoModelForCausalLM; import torch; m = AutoModelForCausalLM.from_pretrained('.\qwen2.5-3b-instruct', torch_dtype=torch.bfloat16, low_cpu_mem_usage=True, local_files_only=True, use_safetensors=True); print('LOADED OK')"
|
| 12 |
-
|
| 13 |
-
Write-Host "--- Exit code ---"
|
| 14 |
-
echo $LASTEXITCODE
|
|
|
|
| 1 |
+
[Environment]::SetEnvironmentVariable("HF_HOME", "D:\hf_cache", "User")
|
| 2 |
+
[Environment]::SetEnvironmentVariable("TRANSFORMERS_CACHE", "D:\hf_cache", "User")
|
| 3 |
+
mkdir D:\hf_cache -Force
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|