Spaces:

syedmudassir16 commited on
Commit
5c39e15
1 Parent(s): 12e88b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -54,6 +54,7 @@ class DocumentRetrievalAndGeneration:
54
  model_id,
55
  torch_dtype=torch.bfloat16,
56
  device_map="auto",
 
57
  quantization_config=quantization_config
58
  )
59
  return tokenizer, model
 
54
  model_id,
55
  torch_dtype=torch.bfloat16,
56
  device_map="auto",
57
+ print("--")
58
  quantization_config=quantization_config
59
  )
60
  return tokenizer, model