Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -64,8 +64,8 @@ emotion_prediction_tokenizer = AutoTokenizer.from_pretrained("bhadresh-savani/di
|
|
64 |
response_model_name = "gpt2-xl"
|
65 |
response_tokenizer = AutoTokenizer.from_pretrained(response_model_name)
|
66 |
with init_empty_weights():
|
67 |
-
response_model = AutoModelForCausalLM.from_pretrained(response_model_name
|
68 |
-
|
69 |
# Set the pad token
|
70 |
response_tokenizer.pad_token = response_tokenizer.eos_token
|
71 |
|
|
|
64 |
response_model_name = "gpt2-xl"
|
65 |
response_tokenizer = AutoTokenizer.from_pretrained(response_model_name)
|
66 |
with init_empty_weights():
|
67 |
+
response_model = AutoModelForCausalLM.from_pretrained(response_model_name)
|
68 |
+
response_model.tie_weights()
|
69 |
# Set the pad token
|
70 |
response_tokenizer.pad_token = response_tokenizer.eos_token
|
71 |
|