akashmishra358 commited on
Commit
daacaf7
·
verified ·
1 Parent(s): 5d1f7bd

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -17,7 +17,7 @@ class RAGModel:
17
  def __init__(self, configs) -> None:
18
  self.configs = configs
19
  self.device = configs["model"]["device"]
20
- model_url = configs["model"]["generation_model"]
21
  # quantization_config = BitsAndBytesConfig(
22
  # load_in_4bit=True, bnb_4bit_compute_dtype=torch.float16
23
  # )
 
17
  def __init__(self, configs) -> None:
18
  self.configs = configs
19
  self.device = configs["model"]["device"]
20
+ model_url = configs["model"]["genration_model"]
21
  # quantization_config = BitsAndBytesConfig(
22
  # load_in_4bit=True, bnb_4bit_compute_dtype=torch.float16
23
  # )