nyarderr commited on
Commit
3bbe154
·
verified ·
1 Parent(s): c1da5b4

Update app/emotion_detector.py

Browse files
Files changed (1) hide show
  1. app/emotion_detector.py +2 -2
app/emotion_detector.py CHANGED
@@ -21,7 +21,7 @@ base_model = AutoModelForCausalLM.from_pretrained(
21
  ## load peft model
22
  model = PeftModel.from_pretrained(
23
  base_model,
24
- LORA_PATH,
25
  torch_dtype=torch.float16,
26
  device_map="auto",
27
  )
@@ -29,7 +29,7 @@ model = PeftModel.from_pretrained(
29
  # Load LoRA adapter
30
 
31
 
32
- model = PeftModel.from_pretrained(base_model, LORA_PATH)
33
  model.eval()
34
 
35
 
 
21
  ## load peft model
22
  model = PeftModel.from_pretrained(
23
  base_model,
24
+ LORA_REPO,
25
  torch_dtype=torch.float16,
26
  device_map="auto",
27
  )
 
29
  # Load LoRA adapter
30
 
31
 
32
+ model = PeftModel.from_pretrained(base_model, LORA_REPO)
33
  model.eval()
34
 
35