SustcZhangYX
/

EnvGPT

Environmental Science

Model card Files Files and versions

SustcZhangYX commited on Jan 1

Commit

7f04e30

·

1 Parent(s): 17da505

Update EnvGPTv2

Files changed (1) hide show

exp.py +1 -1

exp.py CHANGED Viewed

@@ -19,7 +19,7 @@ messages = [
 # Pass top_p and temperature directly in the pipeline call
 outputs = pipeline(
     messages,
-    max_new_tokens=512,
     top_p=0.7,  # Add nucleus sampling
     temperature=0.9,  # Add temperature control
 )

 # Pass top_p and temperature directly in the pipeline call
 outputs = pipeline(
     messages,
+    max_new_tokens=4096,
     top_p=0.7,  # Add nucleus sampling
     temperature=0.9,  # Add temperature control
 )