protgpt3
/

ProtGPT3-1.3B

Text Generation

protein-language-model

protein-generation

mixture-of-experts

text-generation-inference

Model card Files Files and versions

protgpt3 commited on 18 days ago

Commit

1bc8acd

·

verified ·

1 Parent(s): 8f58b22

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -93,7 +93,7 @@ inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
 with torch.no_grad():
     output_ids = model.generate(
-        **inputs,
         max_new_tokens=512,
         do_sample=True,
         temperature=0.8,
@@ -118,7 +118,7 @@ inputs = tokenizer(prefix, return_tensors="pt").to(model.device)
 with torch.no_grad():
     output_ids = model.generate(
-        **inputs,
         max_new_tokens=256,
         do_sample=True,
         temperature=0.8,
@@ -150,7 +150,7 @@ inputs = tokenizer(
 with torch.no_grad():
     output_ids = model.generate(
-        **inputs,
         max_new_tokens=256,
         do_sample=True,
         temperature=0.8,

 with torch.no_grad():
     output_ids = model.generate(
+        inputs["input_ids"],
         max_new_tokens=512,
         do_sample=True,
         temperature=0.8,
 with torch.no_grad():
     output_ids = model.generate(
+        inputs["input_ids"],
         max_new_tokens=256,
         do_sample=True,
         temperature=0.8,
 with torch.no_grad():
     output_ids = model.generate(
+        inputs["input_ids"],
         max_new_tokens=256,
         do_sample=True,
         temperature=0.8,