Spaces:
Runtime error
Runtime error
Add offload folder argument.
Browse files
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()
|