Leonydis137 commited on
Commit
a99ec58
·
verified ·
1 Parent(s): 728e5f6

Update src/utils/shared.py

Browse files
Files changed (1) hide show
  1. src/utils/shared.py +1 -1
src/utils/shared.py CHANGED
@@ -6,4 +6,4 @@ llm_model = AutoModelForCausalLM.from_pretrained(
6
  model_type="llama",
7
  max_new_tokens=256,
8
  temperature=0.7
9
- )
 
6
  model_type="llama",
7
  max_new_tokens=256,
8
  temperature=0.7
9
+ )