Spaces:
Runtime error
Runtime error
Update model.py
Browse files
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"]["
|
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 |
# )
|