Hugging Face's logo

flax-community
/
hubert-dementia-screening

Feature Extraction Transformers JAX hubert Inference Endpoints
Model card Files Community
patrickvonplaten commited on
Commit
47a6a98
•
1 Parent(s): 004e84f

Update run_hubert_classifier.py

Browse files
Files changed (1) hide show
  1. run_hubert_classifier.py +1 -1
run_hubert_classifier.py CHANGED
@@ -2,7 +2,7 @@
2
  from hubert_for_sequence_classification import FlaxHubertForSequenceClassification, FlaxHubertModel
3
  import numpy as np
4
 
5
- # need to do some ugly save/reload because of a bug
6
  model = FlaxHubertModel.from_pretrained("facebook/hubert-large-ll60k", from_pt=True)
7
  model.save_pretrained("./")
8
  model = FlaxHubertForSequenceClassification.from_pretrained("./")
2
  from hubert_for_sequence_classification import FlaxHubertForSequenceClassification, FlaxHubertModel
3
  import numpy as np
4
 
5
+ # need to do some ugly save/reload because of a bug: https://github.com/huggingface/transformers/issues/12532
6
  model = FlaxHubertModel.from_pretrained("facebook/hubert-large-ll60k", from_pt=True)
7
  model.save_pretrained("./")
8
  model = FlaxHubertForSequenceClassification.from_pretrained("./")