chayanbhansali commited on
Commit
16bbc5d
·
verified ·
1 Parent(s): bb99597

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ class RAGChatbot:
17
  # )
18
  self.model = AutoModelForCausalLM.from_pretrained(
19
  model_name,
20
- torch_dtype=torch.bfloat16, ,
21
  device_map="auto"
22
  )
23
 
 
17
  # )
18
  self.model = AutoModelForCausalLM.from_pretrained(
19
  model_name,
20
+ torch_dtype=torch.bfloat16,
21
  device_map="auto"
22
  )
23