Spaces:
Running on CPU Upgrade

ziem-io commited on
Commit
cbe7bbd
·
1 Parent(s): 9a5d8ec

Fix: Quickfix 'init_empty_weights' error

Browse files
Files changed (1) hide show
  1. lib/bert_regressor.py +4 -1
lib/bert_regressor.py CHANGED
@@ -20,7 +20,10 @@ class BertMultiHeadRegressor(nn.Module):
20
  super().__init__()
21
 
22
  # Beliebigen Encoder laden
23
- self.encoder = AutoModel.from_pretrained(pretrained_model_name)
 
 
 
24
  hidden_size = self.encoder.config.hidden_size
25
 
26
  # Erst alles einfrieren …
 
20
  super().__init__()
21
 
22
  # Beliebigen Encoder laden
23
+ self.encoder = AutoModel.from_pretrained(
24
+ pretrained_model_name,
25
+ low_cpu_mem_usage=False # vermeidet accelerate-Abhängigkeit zur Init
26
+ )
27
  hidden_size = self.encoder.config.hidden_size
28
 
29
  # Erst alles einfrieren …