Sephfox commited on
Commit
11a08e7
·
verified ·
1 Parent(s): aa8de66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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, device_map="auto", torch_dtype=torch.float16)
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