Update app.py
Browse files
app.py
CHANGED
|
@@ -68,7 +68,7 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
|
|
| 68 |
from accelerate import init_empty_weights, infer_auto_device_map, load_checkpoint_and_dispatch
|
| 69 |
from accelerate import Accelerator
|
| 70 |
|
| 71 |
-
|
| 72 |
subprocess.run(
|
| 73 |
"pip install psutil",
|
| 74 |
|
|
|
|
| 68 |
from accelerate import init_empty_weights, infer_auto_device_map, load_checkpoint_and_dispatch
|
| 69 |
from accelerate import Accelerator
|
| 70 |
|
| 71 |
+
import torch
|
| 72 |
subprocess.run(
|
| 73 |
"pip install psutil",
|
| 74 |
|