jmcinern commited on
Commit
558f45b
·
verified ·
1 Parent(s): c8de34f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -36,10 +36,14 @@ class ChatBot:
36
  torch_dtype="auto",
37
  low_cpu_mem_usage=True,
38
  )
 
 
 
39
  self.model = PeftModel.from_pretrained(
40
  base_model,
41
  DPO_ADAPTER
42
  )
 
43
  print("Model loaded!")
44
  except Exception as e:
45
  print(f"Error loading model: {e}")
 
36
  torch_dtype="auto",
37
  low_cpu_mem_usage=True,
38
  )
39
+ self.model = base_model
40
+ # Uncomment to use DPO adapter, I suspect it is giving problem.
41
+ '''
42
  self.model = PeftModel.from_pretrained(
43
  base_model,
44
  DPO_ADAPTER
45
  )
46
+ '''
47
  print("Model loaded!")
48
  except Exception as e:
49
  print(f"Error loading model: {e}")