You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
from codetf.models import load_model_pipeline
code_generation_model = load_model_pipeline(model_name="codet5", task="pretrained",
model_type="plus-6B", is_eval=True,
load_in_8bit=True, load_in_4bit=False, weight_sharding=False)
result = code_generation_model.predict(["def find_largest_number_in_list():"])
print(result)
I get the below error.
ValueError: Loading Salesforce/codet5p-6b requires you to execute the configuration file in that repo on your local machine. Make sure you have read the code there to avoid malicious use, then set the option trust_remote_code=True to remove this error.
The Readme example works perfectly fine
The text was updated successfully, but these errors were encountered:
When trying to execute the following code:
I get the below error.
ValueError: Loading Salesforce/codet5p-6b requires you to execute the configuration file in that repo on your local machine. Make sure you have read the code there to avoid malicious use, then set the option
trust_remote_code=Trueto remove this error.
The Readme example works perfectly fine
The text was updated successfully, but these errors were encountered: