Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
11 |
return_dict=True,
|
12 |
load_in_8bit=False,
|
13 |
device_map=torch.device("cuda") if torch.cuda.is_available() else "cpu",
|
14 |
-
load_in_8bit_fp32_cpu_offload=True
|
15 |
)
|
16 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|
17 |
|
|
|
11 |
return_dict=True,
|
12 |
load_in_8bit=False,
|
13 |
device_map=torch.device("cuda") if torch.cuda.is_available() else "cpu",
|
14 |
+
#load_in_8bit_fp32_cpu_offload=True
|
15 |
)
|
16 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|
17 |
|