Chat_with_Meta_llama3_8b

Running

rishikumar20202023 commited on Mar 2

Commit

0659005

verified ·

1 Parent(s): 5ca3ab1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,7 +68,7 @@ subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENT
 from accelerate import init_empty_weights, infer_auto_device_map, load_checkpoint_and_dispatch
 from accelerate import Accelerator
 subprocess.run(
     "pip install psutil",

 from accelerate import init_empty_weights, infer_auto_device_map, load_checkpoint_and_dispatch
 from accelerate import Accelerator
+import torch
 subprocess.run(
     "pip install psutil",