dinhdat1110 commited on
Commit
784741f
·
1 Parent(s): b8f84de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -5
app.py CHANGED
@@ -33,11 +33,6 @@ def model_loading_pipeline():
33
  model=model_id,
34
  model_kwargs={
35
  "torch_dtype": torch.float16,
36
- "load_in_4bits": True,
37
- "quantization_config": BitsAndBytesConfig(
38
- load_in_4bit=True,
39
- bnb_4bit_compute_dtype=torch.float16
40
- ),
41
  },
42
  streamer=streamer,
43
  )
 
33
  model=model_id,
34
  model_kwargs={
35
  "torch_dtype": torch.float16,
 
 
 
 
 
36
  },
37
  streamer=streamer,
38
  )