TeraSpace commited on
Commit
682df44
·
verified ·
1 Parent(s): 27f532f

Accept AutoModel framework loading options

Browse files
Files changed (1) hide show
  1. modeling_teratts.py +15 -0
modeling_teratts.py CHANGED
@@ -48,6 +48,21 @@ class TeraTTSModel(PreTrainedModel):
48
  """
49
  if model_args:
50
  raise TypeError("TeraTTSModel.from_pretrained accepts no positional model arguments")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
  source = Path(pretrained_model_name_or_path)
52
  if source.is_dir():
53
  release = source
 
48
  """
49
  if model_args:
50
  raise TypeError("TeraTTSModel.from_pretrained accepts no positional model arguments")
51
+ # ``AutoModel`` forwards these framework-only values to a custom
52
+ # class after it has already used them. This ONNX runtime has no
53
+ # adapters or PyTorch state dict to load, so safely discard them.
54
+ for key in (
55
+ "trust_remote_code",
56
+ "adapter_kwargs",
57
+ "_from_auto",
58
+ "use_safetensors",
59
+ "weights_only",
60
+ "low_cpu_mem_usage",
61
+ "device_map",
62
+ "torch_dtype",
63
+ "dtype",
64
+ ):
65
+ kwargs.pop(key, None)
66
  source = Path(pretrained_model_name_or_path)
67
  if source.is_dir():
68
  release = source