Commit
·
b97ddc6
1
Parent(s):
40aa7a0
change it back to long
Browse files
app.py
CHANGED
|
@@ -27,8 +27,7 @@ scheduler = LMSDiscreteScheduler(beta_start=beta_start, beta_end=beta_end, beta_
|
|
| 27 |
def text_enc(prompts, maxlen=None):
|
| 28 |
if maxlen is None: maxlen = tokenizer.model_max_length
|
| 29 |
inp = tokenizer(prompts, padding="max_length", max_length=maxlen, truncation=True, return_tensors="pt")
|
| 30 |
-
input_ids = inp.input_ids
|
| 31 |
-
input_ids = input_ids.to(torch.float32) # Convert input_ids to float32 data type
|
| 32 |
return text_encoder(input_ids)[0]
|
| 33 |
|
| 34 |
def do_both(prompts):
|
|
|
|
| 27 |
def text_enc(prompts, maxlen=None):
|
| 28 |
if maxlen is None: maxlen = tokenizer.model_max_length
|
| 29 |
inp = tokenizer(prompts, padding="max_length", max_length=maxlen, truncation=True, return_tensors="pt")
|
| 30 |
+
input_ids = inp.input_ids.to(torch.long)
|
|
|
|
| 31 |
return text_encoder(input_ids)[0]
|
| 32 |
|
| 33 |
def do_both(prompts):
|