Spaces:

binaryMao
/

sira-asr

Sleeping

sira-asr / test_load.py

Antigravity AI

fix: resolve build dependencies and add CORS support

7ef1aa8 7 months ago

1.74 kB

import nemo.collections.asr as nemo_asr
import torch
import os

MODEL_PATH = "c:/Users/USER/queue-buddy/mon_modele_soloni/soloni-114m-tdt-ctc-v3.nemo"

print(f"Testing load from: {MODEL_PATH}")
if not os.path.exists(MODEL_PATH):
    print("Error: File not found!")
    exit(1)

try:
    print("Attempt 1: Generic ASRModel.restore_from")
    model = nemo_asr.models.ASRModel.restore_from(MODEL_PATH, map_location=torch.device('cpu'))
    print("Success Attempt 1")
except Exception as e:
    print(f"Failed Attempt 1: {e}")

try:
    print("\nAttempt 2: EncDecCTCModelBPE.restore_from with strict=False")
    from nemo.collections.asr.models import EncDecCTCModelBPE
    model = EncDecCTCModelBPE.restore_from(MODEL_PATH, map_location=torch.device('cpu'), strict=False)
    print("Success Attempt 2")
except Exception as e:
    print(f"Failed Attempt 2: {e}")

try:
    print("\nAttempt 3: Manual config override")
    from nemo.core.classes import ModelPT
    from omegaconf import OmegaConf
    # We try to extract and modify config
    import tarfile
    import tempfile
    with tempfile.TemporaryDirectory() as tmpdir:
        with tarfile.open(MODEL_PATH, 'r:gz') as tar:
            tar.extract('model_config.yaml', path=tmpdir)
            conf = OmegaConf.load(os.path.join(tmpdir, 'model_config.yaml'))
            # Force remove training data config
            if 'train_ds' in conf: del conf['train_ds']
            if 'validation_ds' in conf: del conf['validation_ds']
            # Re-instantiate from specific class if possible
            # This is complex, but let's see if we can get past abstract method error
            print("Config loaded and cleaned.")
except Exception as e:
    print(f"Failed Attempt 3: {e}")

	import nemo.collections.asr as nemo_asr
	import torch
	import os

	MODEL_PATH = "c:/Users/USER/queue-buddy/mon_modele_soloni/soloni-114m-tdt-ctc-v3.nemo"

	print(f"Testing load from: {MODEL_PATH}")
	if not os.path.exists(MODEL_PATH):
	print("Error: File not found!")
	exit(1)

	try:
	print("Attempt 1: Generic ASRModel.restore_from")
	model = nemo_asr.models.ASRModel.restore_from(MODEL_PATH, map_location=torch.device('cpu'))
	print("Success Attempt 1")
	except Exception as e:
	print(f"Failed Attempt 1: {e}")

	try:
	print("\nAttempt 2: EncDecCTCModelBPE.restore_from with strict=False")
	from nemo.collections.asr.models import EncDecCTCModelBPE
	model = EncDecCTCModelBPE.restore_from(MODEL_PATH, map_location=torch.device('cpu'), strict=False)
	print("Success Attempt 2")
	except Exception as e:
	print(f"Failed Attempt 2: {e}")

	try:
	print("\nAttempt 3: Manual config override")
	from nemo.core.classes import ModelPT
	from omegaconf import OmegaConf
	# We try to extract and modify config
	import tarfile
	import tempfile
	with tempfile.TemporaryDirectory() as tmpdir:
	with tarfile.open(MODEL_PATH, 'r:gz') as tar:
	tar.extract('model_config.yaml', path=tmpdir)
	conf = OmegaConf.load(os.path.join(tmpdir, 'model_config.yaml'))
	# Force remove training data config
	if 'train_ds' in conf: del conf['train_ds']
	if 'validation_ds' in conf: del conf['validation_ds']
	# Re-instantiate from specific class if possible
	# This is complex, but let's see if we can get past abstract method error
	print("Config loaded and cleaned.")
	except Exception as e:
	print(f"Failed Attempt 3: {e}")