Chris4K commited on
Commit
4d00775
·
verified ·
1 Parent(s): 87f1c74

Update services/llama_generator.py

Browse files
Files changed (1) hide show
  1. services/llama_generator.py +2 -1
services/llama_generator.py CHANGED
@@ -43,7 +43,8 @@ class LlamaGenerator(BaseGenerator):
43
  # self.tokenizer = self.load_tokenizer(llama_model_name) # Add this line to initialize the tokenizer
44
 
45
  ):
46
-
 
47
  @observe()
48
  def load_model(self, model_name: str):
49
  # Code to load your model, e.g., Hugging Face's transformers library
 
43
  # self.tokenizer = self.load_tokenizer(llama_model_name) # Add this line to initialize the tokenizer
44
 
45
  ):
46
+ print(llama_model_name)
47
+ print(prm_model_path)
48
  @observe()
49
  def load_model(self, model_name: str):
50
  # Code to load your model, e.g., Hugging Face's transformers library