Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ warnings.filterwarnings("ignore", category=UserWarning)
|
|
45 |
MODEL_NAME = "tiiuae/falcon-7b-instruct"
|
46 |
|
47 |
model = AutoModelForCausalLM.from_pretrained(
|
48 |
-
MODEL_NAME, trust_remote_code=True,
|
49 |
)
|
50 |
model = model.eval()
|
51 |
print('model loadeddddddddddddddddddddddd')
|
|
|
45 |
MODEL_NAME = "tiiuae/falcon-7b-instruct"
|
46 |
|
47 |
model = AutoModelForCausalLM.from_pretrained(
|
48 |
+
MODEL_NAME, trust_remote_code=True, device_map="auto",offload_folder="offload"
|
49 |
)
|
50 |
model = model.eval()
|
51 |
print('model loadeddddddddddddddddddddddd')
|