code
Browse files
app.py
CHANGED
@@ -5,6 +5,8 @@ import gradio as gr
|
|
5 |
import os
|
6 |
from threading import Thread
|
7 |
|
|
|
|
|
8 |
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True'
|
9 |
|
10 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
5 |
import os
|
6 |
from threading import Thread
|
7 |
|
8 |
+
os.system("rm -rf /data-nvme")
|
9 |
+
|
10 |
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'expandable_segments:True'
|
11 |
|
12 |
model = AutoModelForCausalLM.from_pretrained(
|