Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -29,8 +29,7 @@ trained_model = AutoPeftModelForCausalLM.from_pretrained(
|
|
29 |
bnb_4bit_compute_dtype=torch.bfloat16,
|
30 |
bnb_4bit_use_double_quant=True,
|
31 |
bnb_4bit_quant_type='nf4'
|
32 |
-
), torch_dtype=torch.bfloat16,
|
33 |
-
low_cpu_mem_usage=True,
|
34 |
)
|
35 |
tokenizer = AutoTokenizer.from_pretrained(model_token)
|
36 |
|
|
|
29 |
bnb_4bit_compute_dtype=torch.bfloat16,
|
30 |
bnb_4bit_use_double_quant=True,
|
31 |
bnb_4bit_quant_type='nf4'
|
32 |
+
), torch_dtype=torch.bfloat16,offload_folder="offload/",
|
|
|
33 |
)
|
34 |
tokenizer = AutoTokenizer.from_pretrained(model_token)
|
35 |
|