reab5555 commited on
Commit
7c0ef3f
·
verified ·
1 Parent(s): 999d883

Update processing.py

Browse files
Files changed (1) hide show
  1. processing.py +1 -1
processing.py CHANGED
@@ -30,7 +30,7 @@ class LazyPipeline:
30
  @spaces.GPU(duration=250)
31
  def get_pipeline(self):
32
  if self.pipeline is None:
33
- model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"
34
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
35
  model = AutoModelForCausalLM.from_pretrained(
36
  model_name,
 
30
  @spaces.GPU(duration=250)
31
  def get_pipeline(self):
32
  if self.pipeline is None:
33
+ model_name = "google/gemma-2-9b-it"
34
  tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
35
  model = AutoModelForCausalLM.from_pretrained(
36
  model_name,