wisent 0.1.1__py3-none-any.whl → 0.5.2__py3-none-any.whl

wisent-0.5.2.dist-info/RECORD

@@ -0,0 +1,218 @@
+wisent/__init__.py,sha256=isJrmDBLRag7Zc2UK9ZovWGOv7ji1Oh-zJtJMNJFkXw,22
+wisent/benchmarks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/benchmarks/coding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/benchmarks/coding/metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/benchmarks/coding/metrics/evaluator.py,sha256=hI8WvJ8JyDzoKcAi3Yb4QbJ7dn_KsR4rR17k9q93RCs,11892
+wisent/benchmarks/coding/metrics/passk.py,sha256=IZZmGAbvmONPDYX9khzN6tqk6GZe4EvfijSPBWTbrE8,1984
+wisent/benchmarks/coding/metrics/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/benchmarks/coding/metrics/core/atoms.py,sha256=I4wwOUTDjJ_4Jo6Bbpl8Tkv0Uz9zpwNjzeWBrXru-9g,1001
+wisent/benchmarks/coding/output_sanitizer/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/benchmarks/coding/output_sanitizer/cpp_sanitizer.py,sha256=EbTrBgYDAwMJnWiA5PK9Mb7u7s3KtCpk3XQE4MSjdu4,3456
+wisent/benchmarks/coding/output_sanitizer/java_sanitizer.py,sha256=r7IPwM5FDskucjasb8sRtW7vvGZF-uZzaqOLcM3N_ws,4502
+wisent/benchmarks/coding/output_sanitizer/python_sanitizer.py,sha256=2Zifsul69XFQI4_Z5JRbN_pb66qxV51xdth3zGkvQEQ,4873
+wisent/benchmarks/coding/output_sanitizer/utils.py,sha256=FNepdX6bkDbp3unKLsMXNPemwi1iQnhsYEmcLc4SvWI,3153
+wisent/benchmarks/coding/output_sanitizer/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/benchmarks/coding/output_sanitizer/core/atoms.py,sha256=yoPxZg-jA9XE065erGV99Dj6aE3dFvKzogxLo_IWvq8,1185
+wisent/benchmarks/coding/providers/__init__.py,sha256=rpGxV7u93zVQdkXfywxPeq_L5jXNFc0cN1Zj0TsXw2E,731
+wisent/benchmarks/coding/providers/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/benchmarks/coding/providers/core/atoms.py,sha256=6THxuLLpg2GRiKT82ZItyQWgLGx7_Ljzk4YhgjV8zfM,1090
+wisent/benchmarks/coding/providers/livecodebench/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/benchmarks/coding/providers/livecodebench/provider.py,sha256=A_Gz3v7hk1p3Ep3UAG6M-bhB2FMefGybdo8wsp2rdZQ,2449
+wisent/benchmarks/coding/safe_docker/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/benchmarks/coding/safe_docker/entrypoint.py,sha256=--CpgJ9Vn699Jx0WeGUgLEBWlF_f5RBtbo3dLCRJeyM,4277
+wisent/benchmarks/coding/safe_docker/recipes.py,sha256=UWkiBl_k1nmJ6mHB6G0KH6RfLFhV4bH84ITTYmsbT3E,2227
+wisent/benchmarks/coding/safe_docker/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/benchmarks/coding/safe_docker/core/atoms.py,sha256=JE2eOKtiXGLNGSNqaPwtpPbcipoPeDdmVNxgt5_EHYE,3026
+wisent/benchmarks/coding/safe_docker/core/runtime.py,sha256=ACzxKnXNPqA3w5kvllmJduz4ZAH9JR3HYDpHmH-7vM4,4393
+wisent/classifiers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/classifiers/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/classifiers/core/atoms.py,sha256=rHxL6eO3OptQWc8UPKeZCMbnpnXTB_TzHkinMU_rZSk,25972
+wisent/classifiers/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/classifiers/models/logistic.py,sha256=nnE1I4WF9iHNFfeGr5khtoG_cJ4hQlfoE2pa9ZaJlKw,859
+wisent/classifiers/models/mlp.py,sha256=fOWAaqTOYpB5lyEzUOb-BTkpPIpX4QpeYuH2yPU2DMw,1519
+wisent/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/cli/cli_logger.py,sha256=6vuLGPz3wJAf8tHQhuYatRhnFYiUnlPEFy6ThYpqKrs,4477
+wisent/cli/classifiers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/cli/classifiers/classifier_rotator.py,sha256=LAx5hTZ71Q7I4W0uSz9Bn1V4G_cpU3jmvwt160i3zes,5158
+wisent/cli/data_loaders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/cli/data_loaders/data_loader_rotator.py,sha256=q68sQJGHLZQ5iaEr84Uo_NeRMChsQL8kZJ3D6Y2MDF8,4309
+wisent/cli/evaluators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/cli/evaluators/evaluator_rotator.py,sha256=9EDR7zeOwynZvZ-J5h96sYIl9RlKOUihyJ4X5vWf31w,5686
+wisent/cli/steering_methods/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/cli/steering_methods/steering_rotator.py,sha256=_S32ItI4IHNNc230Wr8m1q5gP2BrwexSyN1_nkiUOTI,4212
+wisent/cli/wisent_cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/cli/wisent_cli/main.py,sha256=7_llYPnxVbYCnLU5rIunHNz4jyZdXDJYz8IXkfZDeTI,3503
+wisent/cli/wisent_cli/shell.py,sha256=LVA1hpC7JEUxR_j5p0APPL1jAoyWlt12s4_GMopa1D0,2372
+wisent/cli/wisent_cli/ui.py,sha256=M52163wvh0DgSjsRG69RhqEX87faceXFm-roOn7bH2M,2960
+wisent/cli/wisent_cli/version.py,sha256=ICeC5AhJ79Gx1kISaMY1WQHMl4al3elA4RKLAly-6uc,87
+wisent/cli/wisent_cli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/cli/wisent_cli/commands/help_cmd.py,sha256=C8iA5mMdEBUQovtgJDqAGH6b9TbKSHPhnESQE8_6P5M,1614
+wisent/cli/wisent_cli/commands/listing.py,sha256=1UaKNRrlDOacQSPxp_yu3cAoWpTCwNU1-8DO2oUWJ5w,5552
+wisent/cli/wisent_cli/commands/train_cmd.py,sha256=Pp_DY6P-zk8PvCuz6ds9JxH7BWpqQM_jc_ZfZ1lGIR0,13427
+wisent/cli/wisent_cli/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/cli/wisent_cli/util/aggregations.py,sha256=RW2f-B18k4R0gFlIFACQmrhFKSwfxREUqqu5PaI951Y,1415
+wisent/cli/wisent_cli/util/parsing.py,sha256=DvBTcoBItDGJQI-AE4bs0otBJ7ElLynNhqRDzgeQjzs,4067
+wisent/core/__init__.py,sha256=V48vT6tZj0iwp9rPWuElGJ5pyUyL1foWbQLS3tf4RiE,757
+wisent/core/autonomous_agent.py,sha256=2k1PLWm2DZ6C2fmsjMDKmQ1_wfN7KhpilHrjkTw8nMw,52489
+wisent/core/bigcode_integration.py,sha256=TIaPQDbPRDPdnCq8U-Gwl4lgayPfhOabOVQddqxotY4,19927
+wisent/core/detection_handling.py,sha256=iiuKpzAbJfx_KFn2SFABQHOeeWblDJMXjzGwGDeKqcs,11127
+wisent/core/download_full_benchmarks.py,sha256=KT5zHkau7zk3pgzoCS1UY8Tz7-M9JtlkNXZ9U1uDh2k,55946
+wisent/core/hyperparameter_optimizer.py,sha256=GN-IY27kkFKuXwHbvalfEW5UVf4NfZ0G9rMWK2ThEAY,17319
+wisent/core/lm_eval_harness_ground_truth.py,sha256=JjAbr520h01CcPUEgJNhj6RXWBja2WAvw6Jzq5av7qw,64912
+wisent/core/log_likelihoods_evaluator.py,sha256=meHdfoAlKUm0l8F1Y6aKcBHBi_wNgSS_b8uBJrD9378,15535
+wisent/core/managed_cached_benchmarks.py,sha256=JbvpZ1fgSuQQhyQVKEvqrQZRHGqfnjo9NFhgITFoFsE,22854
+wisent/core/mixed_benchmark_sampler.py,sha256=tKQCHUXVuYeCyx4VZt8O1hGyB-TOY_SQ_SYi8cyApII,13585
+wisent/core/model_config_manager.py,sha256=rQAdSmk3GFlZXyHp3fSV1bORxiZWhmzIz1uo3H4JtkA,12009
+wisent/core/model_persistence.py,sha256=6_vc1Ndujd4v0O68giINSTvYhmb7-AiacWwAbqLOrls,10636
+wisent/core/multi_steering.py,sha256=IpaVrs28tZQM8jXbgUbrhq3bWbH4KaNhUZQcWit7_p0,11711
+wisent/core/parser.py,sha256=_YDeSuQMx0zNknz9rX3Ls1YPT1x5eohoY8rfjeoqxV8,69091
+wisent/core/representation.py,sha256=hBl_N9qbr5Gsa7GCQ0nMWRm82RqYEfhd9cyf0PPH5LY,195
+wisent/core/sample_size_optimizer.py,sha256=6wegGXZpdGpiR4R0YJ1D2JqLr6yinMndEx2gB5FL80s,23666
+wisent/core/sample_size_optimizer_v2.py,sha256=bVYJRZC4_Mrq-HFlYLyv-9tWvqEHJ3kCeIwlmYOwI6I,13286
+wisent/core/save_results.py,sha256=PRwaA5qO6EOsvURvLBl3YhvanlC0D0G4iYqxYAQ7sw8,13737
+wisent/core/steering.py,sha256=GJsARTiaB72fyGQ8UGmTPBmIRb_0OL_-2BXQANSnKqg,22535
+wisent/core/steering_method.py,sha256=-hZqtvwRS7sGqQJUd36MoPm0rjbO1LrtPAYmcIk8BqQ,462
+wisent/core/steering_optimizer.py,sha256=wxa4p4aMjJWOknt2Jph28xPgYoEMxZVmp1GFA1pM3Wk,54759
+wisent/core/task_interface.py,sha256=OlWdcxkprmZcOto-bXmg75kzUcWzH_kyW_e7w2FdPLM,4471
+wisent/core/task_selector.py,sha256=QVgozUuiM74BMUJ8Ucb_sn6HQk5v0wL_QUsqKb55vJE,6224
+wisent/core/time_estimator.py,sha256=DcgSzW-hr9BjmXJwBnGqE2dkFK0zgyz5WNF7934CJ9k,5778
+wisent/core/timing_calibration.py,sha256=4eTn1GWGqzsXmWj9aryXUsZbFEPcJ3IvfCIfJJUFA0w,6860
+wisent/core/user_model_config.py,sha256=8optjLqf9wTDtOf0c705d5_Rr2zE67jV9BNqoY-TRvA,6735
+wisent/core/activations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/activations/activations_collector.py,sha256=wRqtxz5ahVrkd0ja1lAIidf6PL6wxfz9YvR8GT5OqXI,14146
+wisent/core/activations/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/activations/core/atoms.py,sha256=FeeQj-YKb-1rLX0gnzFc5VM47ikWwfFh3datRduj6mQ,9101
+wisent/core/agent/__init__.py,sha256=bIJAwg-AMQD-thjollZPnkA996zkUR21dPM_TS17ptw,478
+wisent/core/agent/budget.py,sha256=9uLBeqmFRvBPGlNxGM0JbEz_mifH3MpLZp8Lw5IwaiM,24386
+wisent/core/agent/device_benchmarks.py,sha256=YWX2_3yzpK26Guz8S9QTuFNvozwYX-tqvhiOleBsBMU,26138
+wisent/core/agent/diagnose.py,sha256=3C5yI_gYd_8pQRdy9617tZzSGqL2rFsmXh1-WXH7Qmk,9312
+wisent/core/agent/steer.py,sha256=qcYMAGTBvPxNLozHkK0uzIKrY1eyK1kCtYx4_wMJOL4,9802
+wisent/core/agent/timeout.py,sha256=I-G4KeAYuz1le6xTLGEWtlOxShWBi6IYSyKuJu-GUz0,4334
+wisent/core/agent/diagnose/__init__.py,sha256=2-4OsA6zqsJKKUmn3okEAXe8Bt4jNUsolIeK-G27f8w,1449
+wisent/core/agent/diagnose/agent_classifier_decision.py,sha256=NgMaogSsXBti8sNWv-Xzna1jwcvZKPsU7M6dX28F6EI,28997
+wisent/core/agent/diagnose/classifier_marketplace.py,sha256=CrXhiZt1XWOc9uuXLdA3AO4qsnVZ92qnVKZ1DFLJTZA,22161
+wisent/core/agent/diagnose/create_classifier.py,sha256=UJKFgjGvIsRKfc5fPkfh227bAWcDD8M4jYNEeHffufI,44182
+wisent/core/agent/diagnose/response_diagnostics.py,sha256=-bNGMq3mapWfZI5SQJAMQHuNRFtc1H6SpTpPtdpCFms,10609
+wisent/core/agent/diagnose/select_classifiers.py,sha256=Uv5dxEmZRjjkZKmVmYQ1jC_XJVyM2MF5dv-4QRuTWKY,18320
+wisent/core/agent/diagnose/synthetic_classifier_option.py,sha256=p_za96LiSXDg0uxvypG0dPMWqirA2srYFpkRTLGCR0Q,32483
+wisent/core/agent/diagnose/test_synthetic_classifier.py,sha256=EWPPHtoND797XfoUjNZcmcuBp8w3PYm4YIBu1JXhzPE,2688
+wisent/core/agent/diagnose/tasks/__init__.py,sha256=rfImwPtKCAfz-ASOBQyF4DEhU6hgCuiZBcqh1AaUE80,704
+wisent/core/agent/diagnose/tasks/task_manager.py,sha256=Nd3ht1vSKmiTxxQx5aCe9QTH4yJwGvLdoUKlimZTfAg,62764
+wisent/core/agent/diagnose/tasks/task_relevance.py,sha256=D4UBr0TqUNXkDZnNgA5wa4NYHSKtDaiugYeVg5zGQjs,3250
+wisent/core/agent/diagnose/tasks/task_selector.py,sha256=ll34stireeqW-B_T4daf_91kujzVFQ8sOilk-JrxpHA,5414
+wisent/core/contrastive_pairs/__init__.py,sha256=AbaAf-t_nyVVy_vLjp8WAlMDmNun3KNp_GMWAK25r9g,429
+wisent/core/contrastive_pairs/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/contrastive_pairs/core/atoms.py,sha256=_zghw6c8iisW_SqBIUCoAnzhc5q7t5EgZ4zzTPxeLwQ,1129
+wisent/core/contrastive_pairs/core/buliders.py,sha256=VWe4StOd3SsV0FBzg8b_2KG_ARiIITrwkfHR5xZNBxk,1975
+wisent/core/contrastive_pairs/core/pair.py,sha256=VsoEXnZ8wpwuictZRpjp1O8pTeMnHvIn-jn01NIA5CI,8095
+wisent/core/contrastive_pairs/core/response.py,sha256=bEUqW53jE7uGEdHFsu3pw209aHxcAORykYcXW-JdWUY,6147
+wisent/core/contrastive_pairs/core/serialization.py,sha256=zpF5BZMFYb-ltD69Xuy53w7qnQ1b5OdkGNVR4pfWbZ8,11157
+wisent/core/contrastive_pairs/core/set.py,sha256=8N1jN0bbuklkhhcDegd0L6Oc9Tv9fFFfoP1LlZSWVNU,4624
+wisent/core/contrastive_pairs/diagnostics/__init__.py,sha256=aAtEI-oOmR2rTmcpMynODIjuTf7S8dW6wmyQS01aUfY,1498
+wisent/core/contrastive_pairs/diagnostics/activations.py,sha256=TUNxU-HV3oeQxFbZhKo-OISM4mzR-Bib0naHmbqWzk8,1736
+wisent/core/contrastive_pairs/diagnostics/base.py,sha256=uBi8PdTd6BRyy0lmGjAZLTZdgiiWwPNtsmKkBFCmlD0,2658
+wisent/core/contrastive_pairs/diagnostics/control_vectors.py,sha256=ypRTEpVzSQDXEqyN-t7ssMoEcM44W8qABwImETEYO9o,6037
+wisent/core/contrastive_pairs/diagnostics/coverage.py,sha256=MpT6_IdLqtMpav6mOCiNuemBVFvxWzkUbj7j3ZNx-48,2761
+wisent/core/contrastive_pairs/diagnostics/divergence.py,sha256=Io3AcGluJogz4qENWu0ivQyFR_5bLN49BzCTI7DIVa4,3430
+wisent/core/contrastive_pairs/diagnostics/duplicates.py,sha256=uwL-RCoQK6e791drLCSl-_z-rYAZXJYXX23zNYNCY0Y,4552
+wisent/core/contrastive_pairs/lm_eval_pairs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/contrastive_pairs/lm_eval_pairs/atoms.py,sha256=iSvDrDlwvhdD8T6aT1NylUiyu8qRCdZzrMFWw1j7Sjg,8575
+wisent/core/contrastive_pairs/lm_eval_pairs/lm_extractor_manifest.py,sha256=gfovSFBdTISyJyw1CXwi57_O15OESmB-AOuh5q6ENMQ,298
+wisent/core/contrastive_pairs/lm_eval_pairs/lm_extractor_registry.py,sha256=DQn4bKRIC6fSpXvTzXxwCcrVe3EKZZesBG30qTCWeHc,4388
+wisent/core/contrastive_pairs/lm_eval_pairs/lm_task_pairs_generation.py,sha256=Q128ATWcfE6NaQOm0O5oMtVRZbXzkeJMlkfCiGNkjyo,1657
+wisent/core/contrastive_pairs/lm_eval_pairs/lm_task_extractors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/contrastive_pairs/lm_eval_pairs/lm_task_extractors/winogrande.py,sha256=rdBYDVlOifA4m5AXLTp9_ttkn2MFs9Yulg371WOjL1s,4044
+wisent/core/data_loaders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/data_loaders/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/data_loaders/core/atoms.py,sha256=sjSS5Eb136hE29dHpKaMA0ohADc0k2iBj_PjpC40GuM,3470
+wisent/core/data_loaders/loaders/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/data_loaders/loaders/custom.py,sha256=Xe1sOHH3_dRjBnQg9vwMM-XA8ROn65dUr9TeT-nuNtQ,4144
+wisent/core/data_loaders/loaders/lm_loader.py,sha256=3ZLi3UnoePd0YgubMlT4CTVWtfpbEHVfvi8QKVq0-Zw,7812
+wisent/core/evaluators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/evaluators/oracles/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/evaluators/oracles/interactive.py,sha256=f3v2_N17fKzGyeOxONRJbrbn8i5uMeZmRvMmF0ShNf0,2638
+wisent/core/evaluators/oracles/nlp_evaluator.py,sha256=KxbnF-I2IFbBQpoYyjQKGbYh4NErsEuhTCRYX_Tob8o,18220
+wisent/core/evaluators/oracles/user_specified.py,sha256=V1dKrNj3Oq7UC_I7DT0WGnktP7R_DSW6UAwDdrA8SnE,2360
+wisent/core/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/models/wisent_model.py,sha256=yJBcz3GjR7O-ySTV2vvOsOrL9xDvXsG0W9Gr0HR_0sc,28729
+wisent/core/models/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/models/core/atoms.py,sha256=_Bpz0Sfiq6_VswThIltUwNGj_ukl5MhAg8RrgMKwEBM,15756
+wisent/core/optuna/__init__.py,sha256=sTfwRnrRyKrCNVsF_qCjBDFEZC0ZmUZ7m6IE0iHfTVs,1914
+wisent/core/optuna/classifier/__init__.py,sha256=vv2wCAbw8ccZxq3vxrQt9plUdbr7VJj-t2rRnh6jBR8,819
+wisent/core/optuna/classifier/activation_generator.py,sha256=K384F_r65z-9f1scr6DoNNzlD1OCVEY9W5TxXilDM5E,14331
+wisent/core/optuna/classifier/classifier_cache.py,sha256=Ng6M8MZoMQx0SpoeVb1ZAmQcVgxzEhlVnDM5RSp1eFo,17380
+wisent/core/optuna/classifier/optuna_classifier_optimizer.py,sha256=OHcUE1Kl_KY0X3Kqp0K_VBIeptp_tqqpCnFVNil9Q_E,22392
+wisent/core/optuna/steering/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/optuna/steering/bigcode_evaluator_wrapper.py,sha256=e3rdGkYPcvRoBzCPpN7xfL7T6Vbt3X9LCLTA39jC23A,6641
+wisent/core/optuna/steering/data_utils.py,sha256=adgA8gXaHLbllhOKspmPxe0tjRZKrc4M8tqs7IQDrK4,12195
+wisent/core/optuna/steering/metrics.py,sha256=Tf4WC3jaVcke5yFMLkBfbc8fhA96TdnL4hXXfGgJ8WM,20378
+wisent/core/optuna/steering/optuna_pipeline.py,sha256=i71osz12k--TGJzd33eH7J0ngQnEJm7gW1LBSS2PBg0,74792
+wisent/core/optuna/steering/steering_optimization.py,sha256=JNtgIWD4b5D7a_VTVlVf7M-yGBItp9p_6g73UhBsGp4,45495
+wisent/core/prompts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/prompts/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/prompts/core/atom.py,sha256=yb3uF7a8Vu7QL_UssriwHlIiqDef43NmppFg7Yj_JTg,1519
+wisent/core/prompts/core/prompt_formater.py,sha256=V-wgz8OPya550p_rkbOkiH-QBzjF3xmnqafCEcNMnuY,5671
+wisent/core/prompts/prompt_stratiegies/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/prompts/prompt_stratiegies/direct_completion.py,sha256=CZq2P1CIyg7yOUS9rwv2KL-lEZDzyrAO4f95lNsujgA,794
+wisent/core/prompts/prompt_stratiegies/instruction_following.py,sha256=ZZ4jM7yo5xZn4-QUt-tFDr37eiZK9TBy5Ygl6xdXP_g,802
+wisent/core/prompts/prompt_stratiegies/multiple_choice.py,sha256=vWYEjvm8zt8g1f1hH9t_hgFhmb-kyJSvsxtZWpSRe-8,867
+wisent/core/prompts/prompt_stratiegies/role_playing.py,sha256=pWNopxpziTGf3hrmBkTuLm1HNKb_pnnka2MTdcNWOEY,1066
+wisent/core/steering_methods/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/steering_methods/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/steering_methods/core/atoms.py,sha256=v8DcO4SSmJXv8KFaYs-A4nR4D2is7Q5SEdFJlZ2SCTg,5420
+wisent/core/steering_methods/methods/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/steering_methods/methods/caa.py,sha256=CsMf1QvyywOr0nWehDQDn17IKAZ3qysURwPP6Jgbu5w,1683
+wisent/core/tasks/__init__.py,sha256=AjsAd4VhMOh2c0_y1ZRYhNsc39cT1lec7M8yHKvua84,6727
+wisent/core/tasks/aime_task.py,sha256=SLGcmf_ywtI73JeqQNjuuKVWgddHNmqqq2jhnQS10vw,5417
+wisent/core/tasks/file_task.py,sha256=DUMhrMFXHXfcuNsPg-LbnCg8aL8fLgdooboZCTfdgSc,7079
+wisent/core/tasks/hle_task.py,sha256=YBSvAA4WnM4OXdsAz6ryh6fL41vPwVHrMHF5OJeLkE4,7043
+wisent/core/tasks/hmmt_task.py,sha256=k-mIWsSl3IEGfGXIPigpw21K2CTMqQ47QDnX__rVQhQ,4742
+wisent/core/tasks/livecodebench_task.py,sha256=BjVn1IC9Hzsd7ADM7v_KYYcJCQ6syZX8BhP1wKHncMo,9235
+wisent/core/tasks/livemathbench_task.py,sha256=G--Gl_1wczDZgB6-wCvvBbXzrOBo9IDbSUqkJxn82Y8,6407
+wisent/core/tasks/lm_eval_task.py,sha256=FfyiJ0kPvvqef81c65Rbdx65CbsucupVC8iKwYHQxb4,14336
+wisent/core/tasks/math500_task.py,sha256=e_qpzG_ppuZxoq3Vi40n_9-vOCf4lRAWO4zIiyUezgU,3124
+wisent/core/tasks/polymath_task.py,sha256=SNZELmzdQM93rjN2_zdnqXTVokDnBFIj-cXE4dsKSTI,5933
+wisent/core/tasks/supergpqa_task.py,sha256=_axLCD7-ZoigxTfaYwVrzyHuo5woeJxKaBpWSYJZ1Tg,9069
+wisent/core/tracking/__init__.py,sha256=Ey03RCi2x-hPTXFjUGum3qvHWF9D-BpV0B5j0St5giw,1156
+wisent/core/tracking/latency.py,sha256=YWvotesKW1k_-7xmXvFQucOQJq3Uu87M_FkudOjynTE,21855
+wisent/core/tracking/memory.py,sha256=fSGrZzHXvOFz3X0V_80CPvSLT5cQeMWW4OT2QcunlA0,13087
+wisent/core/trainers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/core/trainers/steering_trainer.py,sha256=JAdYwPhiakqN70WLqpZFZBsGHu2yed5v-TSM7ppjKJc,10557
+wisent/core/trainers/core/__init__.py,sha256=D0JX0-XCHdtLrCXhVDHNQafvyWCvJ4-o4UKtkH1lI1k,1257
+wisent/core/trainers/core/atoms.py,sha256=ycWk0G-7EIAEOnQL-o5_V5B8KTQ7CQUilGF4ibjighM,1536
+wisent/opti/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/opti/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/opti/core/atoms.py,sha256=9UZeb_SOdDxQ6FBhdAf9qXaEXElImKUsoAMsV0c4yZg,5266
+wisent/opti/methods/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/opti/methods/opti_classificator.py,sha256=U7F3y6weqe6d9SdhR_w2KyGuu_8X27bkuCn4lHrVCw8,6842
+wisent/opti/methods/opti_steering.py,sha256=Xh1GZKW6DOobJk88txEw7EQaHU9tVWqnyJOqPe8VvAo,5376
+wisent/synthetic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/cleaners/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/cleaners/deduper_cleaner.py,sha256=CjURqXjh0j1ESJD6-cDoxmAmwC2cWVHDt6j8f7-ePSM,1556
+wisent/synthetic/cleaners/pairs_cleaner.py,sha256=c6UzgbGL-8np9GA8EnX3zZGSExdVcMzIntGtNtGm2Ak,3835
+wisent/synthetic/cleaners/refusaler_cleaner.py,sha256=VdKSxk9Fm-c4GY0vgcyvW28OPz3iNpfKnzPSNDywih8,5350
+wisent/synthetic/cleaners/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/cleaners/core/atoms.py,sha256=XUrgDhrPs77gCWq9y4VSDaC4WlylRrT1o9vrZjtpB7Y,1350
+wisent/synthetic/cleaners/methods/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/cleaners/methods/base_dedupers.py,sha256=-FBpIA-EDQ5EqIcYz8oYN7yD2s-7kxOR25MqGE1KmP0,11698
+wisent/synthetic/cleaners/methods/base_refusalers.py,sha256=a6a5kyXlsJESLYBI0LOUVNVAS1ObOJEONcmF9Leugyo,10324
+wisent/synthetic/cleaners/methods/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/cleaners/methods/core/atoms.py,sha256=kxD-CnS-u2GXAiezNYEQJ0asgxbMpmR4sXroSz5QjxE,1563
+wisent/synthetic/db_instructions/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/db_instructions/mini_dp.py,sha256=HgRDwofFdvRyQnTfFwGiCokP3ret03GrQEI13gb7V9I,1780
+wisent/synthetic/db_instructions/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/db_instructions/core/atoms.py,sha256=jBazuD37hyBMIsbFNDjpwlGnxECAOuh4YT0fmpH8C6o,748
+wisent/synthetic/generators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/generators/pairs_generator.py,sha256=RfpPdBTqNyWHtYS3ZT4WX_0h5fXLDSDTLXtoRXRcp2k,6551
+wisent/synthetic/generators/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/generators/core/atoms.py,sha256=9wL0v38BCqn3y9LtoRkQsK_X3egjdYcPmFXH0mgFSWg,2290
+wisent/synthetic/generators/diversities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/generators/diversities/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/generators/diversities/core/core.py,sha256=TjSj5T7NE5kRH-ABcFqb1Hz_j3Z6F_TcV-95uHD5Xw8,2201
+wisent/synthetic/generators/diversities/methods/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wisent/synthetic/generators/diversities/methods/fast_diversity.py,sha256=Z2UzTbzyJFM_ToxCoXM_LQQQ1Jc6BZknrbpikTG1MRw,8522
+wisent-0.5.2.dist-info/licenses/LICENSE,sha256=wy0iaw8b2tyqZAfKHib3lP3PJ9o88FDCg92oUHh3sDQ,1073
+wisent-0.5.2.dist-info/METADATA,sha256=IRtye3gZz9T6U3O0fFUap_xvO1rcnnE52hWlIKvPq4g,2424
+wisent-0.5.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+wisent-0.5.2.dist-info/top_level.txt,sha256=2Ts9Iyldnb3auIN2HBBaHPknRy7nSRDm2f6RGzYgr8A,7
+wisent-0.5.2.dist-info/RECORD,,

{wisent-0.1.1.dist-info → wisent-0.5.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.8.2)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

{wisent-0.1.1.dist-info → wisent-0.5.2.dist-info/licenses}/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2024 Wisent Team
+Copyright (c) 2023-2024 Wisent Team
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
@@ -18,4 +18,4 @@ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.
+SOFTWARE. 

wisent/activations/__init__.py

@@ -1,9 +0,0 @@
-"""
-Functionality for extracting and managing model activations.
-"""
-
-from wisent.activations.client import ActivationsClient
-from wisent.activations.extractor import ActivationExtractor
-from wisent.activations.models import Activation, ActivationBatch
-
-__all__ = ["ActivationsClient", "ActivationExtractor", "Activation", "ActivationBatch"] 

wisent/activations/client.py

@@ -1,97 +0,0 @@
-"""
-Client for interacting with the activations API.
-"""
-
-from typing import Dict, List, Optional, Union
-
-from wisent.activations.extractor import ActivationExtractor
-from wisent.activations.models import Activation, ActivationBatch
-from wisent.utils.auth import AuthManager
-from wisent.utils.http import HTTPClient
-
-
-class ActivationsClient:
-    """
-    Client for interacting with the activations API.
-    
-    Args:
-        auth_manager: Authentication manager
-        base_url: Base URL for the API
-        timeout: Request timeout in seconds
-    """
-    
-    def __init__(self, auth_manager: AuthManager, base_url: str, timeout: int = 60):
-        self.auth_manager = auth_manager
-        self.http_client = HTTPClient(base_url, auth_manager.get_headers(), timeout)
-    
-    def extract(
-        self,
-        model_name: str,
-        prompt: str,
-        layers: Optional[List[int]] = None,
-        tokens_to_extract: Optional[List[int]] = None,
-        device: Optional[str] = None,
-    ) -> ActivationBatch:
-        """
-        Extract activations from a model for a given prompt.
-        
-        Args:
-            model_name: Name of the model
-            prompt: Input prompt
-            layers: List of layers to extract activations from (default: [-1])
-            tokens_to_extract: List of token indices to extract (default: [-1])
-            device: Device to use for extraction (default: "cuda" if available, else "cpu")
-            
-        Returns:
-            Batch of activations
-        """
-        extractor = ActivationExtractor(model_name, device=device)
-        return extractor.extract(prompt, layers, tokens_to_extract)
-    
-    def upload(self, batch: ActivationBatch) -> Dict:
-        """
-        Upload a batch of activations to the Wisent backend.
-        
-        Args:
-            batch: Batch of activations
-            
-        Returns:
-            Response from the API
-        """
-        return self.http_client.post("/activations/upload", json_data=batch.to_dict())
-    
-    def get(self, batch_id: str) -> ActivationBatch:
-        """
-        Get a batch of activations from the Wisent backend.
-        
-        Args:
-            batch_id: ID of the batch
-            
-        Returns:
-            Batch of activations
-        """
-        data = self.http_client.get(f"/activations/{batch_id}")
-        return ActivationBatch(**data)
-    
-    def list(
-        self,
-        model_name: Optional[str] = None,
-        limit: int = 100,
-        offset: int = 0,
-    ) -> List[Dict]:
-        """
-        List activation batches from the Wisent backend.
-        
-        Args:
-            model_name: Filter by model name
-            limit: Maximum number of results
-            offset: Offset for pagination
-            
-        Returns:
-            List of activation batch metadata
-        """
-        params = {"limit": limit, "offset": offset}
-        if model_name:
-            params["model_name"] = model_name
-            
-        return self.http_client.get("/activations", params=params) 

wisent/activations/extractor.py

@@ -1,251 +0,0 @@
-"""
-Functionality for extracting activations from models.
-"""
-
-import logging
-from typing import Dict, List, Optional, Tuple, Union
-
-import torch
-from torch.utils.hooks import RemovableHandle
-from transformers import AutoModelForCausalLM, AutoTokenizer
-
-from wisent.activations.models import Activation, ActivationBatch, ActivationExtractorConfig
-
-logger = logging.getLogger(__name__)
-
-
-class ActivationExtractor:
-    """
-    Extracts activations from transformer models.
-    
-    Args:
-        model_name: Name of the model to extract activations from
-        config: Configuration for extraction
-        device: Device to use for extraction
-    """
-    
-    def __init__(
-        self,
-        model_name: str,
-        config: Optional[ActivationExtractorConfig] = None,
-        device: Optional[str] = None,
-    ):
-        self.model_name = model_name
-        self.config = config or ActivationExtractorConfig()
-        
-        if device:
-            self.config.device = device
-            
-        self.device = self.config.device
-        self.model = None
-        self.tokenizer = None
-        self._hooks = []
-        self._activations = {}
-        
-        logger.info(f"Initializing ActivationExtractor for model {model_name} on {self.device}")
-    
-    def _load_model(self) -> None:
-        """Load the model and tokenizer."""
-        if self.model is None:
-            logger.info(f"Loading model {self.model_name}")
-            self.model = AutoModelForCausalLM.from_pretrained(
-                self.model_name, 
-                torch_dtype=torch.float16 if self.device == "cuda" else torch.float32,
-                device_map=self.device
-            )
-            self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
-            logger.info(f"Model loaded successfully")
-    
-    def _register_hooks(self, layers: List[int]) -> None:
-        """
-        Register hooks to capture activations from specified layers.
-        
-        Args:
-            layers: List of layer indices to capture
-        """
-        self._remove_hooks()
-        self._activations = {}
-        
-        # Get all transformer layers
-        if hasattr(self.model, "transformer"):
-            transformer_layers = self.model.transformer.h
-        elif hasattr(self.model, "model") and hasattr(self.model.model, "layers"):
-            transformer_layers = self.model.model.layers
-        else:
-            raise ValueError(f"Unsupported model architecture: {self.model_name}")
-        
-        num_layers = len(transformer_layers)
-        
-        # Resolve negative indices
-        resolved_layers = []
-        for layer in layers:
-            if layer < 0:
-                resolved_layer = num_layers + layer
-            else:
-                resolved_layer = layer
-                
-            if 0 <= resolved_layer < num_layers:
-                resolved_layers.append(resolved_layer)
-            else:
-                logger.warning(f"Layer index {layer} out of range (0-{num_layers-1}), skipping")
-        
-        # Register hooks for each layer
-        for layer_idx in resolved_layers:
-            layer = transformer_layers[layer_idx]
-            
-            # Define hook function to capture activations
-            def hook_fn(module, input, output, layer_idx=layer_idx):
-                # For most models, output is a tuple with hidden states as the first element
-                if isinstance(output, tuple):
-                    hidden_states = output[0]
-                else:
-                    hidden_states = output
-                
-                if layer_idx not in self._activations:
-                    self._activations[layer_idx] = []
-                
-                # Store a copy of the hidden states
-                self._activations[layer_idx].append(hidden_states.detach())
-            
-            # Register hook on the output of the layer
-            if hasattr(layer, "output"):
-                handle = layer.output.register_forward_hook(
-                    lambda module, input, output, layer_idx=layer_idx: hook_fn(module, input, output, layer_idx)
-                )
-            else:
-                handle = layer.register_forward_hook(
-                    lambda module, input, output, layer_idx=layer_idx: hook_fn(module, input, output, layer_idx)
-                )
-            
-            self._hooks.append(handle)
-            
-        logger.info(f"Registered hooks for layers: {resolved_layers}")
-    
-    def _remove_hooks(self) -> None:
-        """Remove all registered hooks."""
-        for hook in self._hooks:
-            hook.remove()
-        self._hooks = []
-    
-    def _get_token_indices(self, tokens_to_extract: List[int], total_tokens: int) -> List[int]:
-        """
-        Resolve token indices, handling negative indices.
-        
-        Args:
-            tokens_to_extract: List of token indices to extract
-            total_tokens: Total number of tokens
-            
-        Returns:
-            List of resolved token indices
-        """
-        resolved_indices = []
-        
-        for idx in tokens_to_extract:
-            if idx < 0:
-                resolved_idx = total_tokens + idx
-            else:
-                resolved_idx = idx
-                
-            if 0 <= resolved_idx < total_tokens:
-                resolved_indices.append(resolved_idx)
-            else:
-                logger.warning(f"Token index {idx} out of range (0-{total_tokens-1}), skipping")
-        
-        return resolved_indices
-    
-    def extract(
-        self,
-        prompt: str,
-        layers: Optional[List[int]] = None,
-        tokens_to_extract: Optional[List[int]] = None,
-    ) -> ActivationBatch:
-        """
-        Extract activations from the model for a given prompt.
-        
-        Args:
-            prompt: Input prompt
-            layers: List of layers to extract activations from (default: from config)
-            tokens_to_extract: List of token indices to extract (default: from config)
-            
-        Returns:
-            Batch of activations
-        """
-        try:
-            self._load_model()
-            
-            layers = layers or self.config.layers
-            tokens_to_extract = tokens_to_extract or self.config.tokens_to_extract
-            
-            # Register hooks for the specified layers
-            self._register_hooks(layers)
-            
-            # Tokenize the input
-            inputs = self.tokenizer(prompt, return_tensors="pt")
-            input_ids = inputs.input_ids.to(self.device)
-            
-            # Get the total number of tokens
-            total_tokens = input_ids.shape[1]
-            
-            # Resolve token indices
-            token_indices = self._get_token_indices(tokens_to_extract, total_tokens)
-            
-            # Run the model to capture activations
-            with torch.no_grad():
-                self.model(input_ids)
-            
-            # Process captured activations
-            activations = []
-            
-            for layer_idx, layer_activations in self._activations.items():
-                # Layer activations should have shape [batch_size, seq_len, hidden_dim]
-                hidden_states = layer_activations[0]
-                
-                # Get token strings for the specified indices
-                token_strings = {}
-                for token_idx in token_indices:
-                    token_id = input_ids[0, token_idx].item()
-                    token_strings[token_idx] = self.tokenizer.decode([token_id])
-                
-                # Extract activations for the specified tokens
-                for token_idx in token_indices:
-                    # Extract the activation for this token
-                    token_activation = hidden_states[0, token_idx, :].cpu()
-                    
-                    # Create an Activation object
-                    activation = Activation(
-                        model_name=self.model_name,
-                        layer=layer_idx,
-                        token_index=token_idx,
-                        values=token_activation,
-                        token_str=token_strings.get(token_idx)
-                    )
-                    
-                    activations.append(activation)
-            
-            # Clean up
-            self._remove_hooks()
-            
-            # Create and return the batch
-            return ActivationBatch(
-                model_name=self.model_name,
-                prompt=prompt,
-                activations=activations,
-                metadata={"total_tokens": total_tokens}
-            )
-            
-        except Exception as e:
-            logger.error(f"Error extracting activations: {str(e)}")
-            self._remove_hooks()
-            raise
-    
-    def __del__(self):
-        """Clean up resources."""
-        self._remove_hooks()
-        
-        # Free GPU memory
-        if self.model is not None and hasattr(self.model, "to"):
-            self.model = self.model.to("cpu")
-            
-        # Clear CUDA cache
-        if torch.cuda.is_available():
-            torch.cuda.empty_cache()