ogx786 commited on
Commit
9404de3
·
verified ·
1 Parent(s): 4400baa

Update test.txt

Browse files
Files changed (1) hide show
  1. test.txt +3 -14
test.txt CHANGED
@@ -1,14 +1,3 @@
1
- Write-Host "--- RAM ---"
2
- Get-CimInstance Win32_OperatingSystem | Select-Object TotalVisibleMemorySize,FreePhysicalMemory
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