Vipitis commited on
Commit
837e020
1 Parent(s): ab88ade

changing the default model (linked on space too)

Browse files
Files changed (1) hide show
  1. ShaderEval.py +1 -1
ShaderEval.py CHANGED
@@ -212,7 +212,7 @@ class Suite(evaluate.EvaluationSuite):
212
 
213
  # from: https://github.com/huggingface/evaluate/blob/v0.4.0/src/evaluate/evaluation_suite/__init__.py#LL103C5-L129C27
214
  def run(
215
- self, model_or_pipeline: Union[str, "Pipeline", Callable, "PreTrainedModel", "TFPreTrainedModel"] = "Vipitis/CodeGPT-small-java-adaptedGPT2-transfer-shadertoys", #not so useful default model?
216
  snippet: int = "" # noqa: F821
217
  ) -> Dict[str, float]:
218
 
 
212
 
213
  # from: https://github.com/huggingface/evaluate/blob/v0.4.0/src/evaluate/evaluation_suite/__init__.py#LL103C5-L129C27
214
  def run(
215
+ self, model_or_pipeline: Union[str, "Pipeline", Callable, "PreTrainedModel", "TFPreTrainedModel"] = "Vipitis/santacoder-finetuned-Shadertoys-fine", #not so useful default model?
216
  snippet: int = "" # noqa: F821
217
  ) -> Dict[str, float]:
218