generadaidemo commited on
Commit
7fee471
·
1 Parent(s): b410afb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ model = AutoModelForCausalLM.from_pretrained(
8
  config.base_model_name_or_path,
9
  return_dict=True,
10
  device_map="auto"
11
- )
12
  tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
13
 
14
  # Load the Lora model
 
8
  config.base_model_name_or_path,
9
  return_dict=True,
10
  device_map="auto"
11
+ )
12
  tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
13
 
14
  # Load the Lora model