feat: 모델 설정에 양자화 및 메모리 제한 구성 추가
This commit is contained in:
@@ -14,5 +14,14 @@
|
||||
"local_storage_path": "./collected_data",
|
||||
"file_format": "json",
|
||||
"drive_mount_path": "/content/drive/MyDrive/model_Dev/data"
|
||||
},
|
||||
"model_settings": {
|
||||
"use_quantization": true,
|
||||
"quantization_bits": 8,
|
||||
"torch_dtype": "auto",
|
||||
"max_memory": {
|
||||
"gpu": "20GB",
|
||||
"cpu": "60GB"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user