huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-05-04 08:42:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(neuron): remove erroneous line

Browse Source
This commit is contained in:
David Corvoysier 2025-02-25 09:21:25 +00:00
parent f6859c4179
commit e783f88dc5
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
integration-tests/fixtures/neuron/export_models.py
View File

@ -190,7 +190,7 @@ def maybe_export_model(config_name, model_config):
logger.debug("Build logs %s", logs) logger.debug("Build logs %s", logs)
try: try:
container = client.containers.run( client.containers.run(
export_image, export_image,
environment=env, environment=env,
auto_remove=True, auto_remove=True,
@ -199,7 +199,6 @@ def maybe_export_model(config_name, model_config):
shm_size="1G", shm_size="1G",
) )
logger.info(f"Successfully exported model for config {config_name}") logger.info(f"Successfully exported model for config {config_name}")
container.logs()
except Exception as e: except Exception as e:
logger.exception(f"An exception occurred while running container: {e}.") logger.exception(f"An exception occurred while running container: {e}.")
pass pass