Spaces:
Running
Running
Commit
·
a1d6a9e
1
Parent(s):
5e95677
Fuck padding
Browse files- models/gpt2.py +1 -1
models/gpt2.py
CHANGED
@@ -12,7 +12,7 @@ def load():
|
|
12 |
|
13 |
def generate(input_text):
|
14 |
# Tokenize the input text
|
15 |
-
input_ids = tokenizer.encode(input_text, return_tensors="pt", truncation=True
|
16 |
attention_mask = tf.ones_like(input_ids)
|
17 |
|
18 |
# Generate output using the model
|
|
|
12 |
|
13 |
def generate(input_text):
|
14 |
# Tokenize the input text
|
15 |
+
input_ids = tokenizer.encode(input_text, return_tensors="pt", truncation=True)
|
16 |
attention_mask = tf.ones_like(input_ids)
|
17 |
|
18 |
# Generate output using the model
|