ClownRat commited on
Commit
6bc6b84
•
1 Parent(s): b0e8c0a

Add offload folder argument.

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -66,7 +66,8 @@ class Chat:
66
  self.tokenizer, self.model, processor, context_len = load_pretrained_model(
67
  model_path, model_base, model_name,
68
  load_8bit, load_4bit,
69
- device=device)
 
70
  self.processor = processor
71
  self.conv_mode = conv_mode
72
  self.conv = conv_templates[conv_mode].copy()
 
66
  self.tokenizer, self.model, processor, context_len = load_pretrained_model(
67
  model_path, model_base, model_name,
68
  load_8bit, load_4bit,
69
+ device=device,
70
+ offload_folder="save_folder")
71
  self.processor = processor
72
  self.conv_mode = conv_mode
73
  self.conv = conv_templates[conv_mode].copy()