Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -6,10 +6,12 @@ from threading import Thread
|
|
6 |
import torch
|
7 |
import spaces
|
8 |
import gradio as gr
|
|
|
9 |
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, TextIteratorStreamer
|
10 |
|
11 |
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
12 |
|
|
|
13 |
MODEL_ID = os.environ.get("MODEL_ID")
|
14 |
CHAT_TEMPLATE = os.environ.get("CHAT_TEMPLATE")
|
15 |
MODEL_NAME = MODEL_ID.split("/")[-1]
|
|
|
6 |
import torch
|
7 |
import spaces
|
8 |
import gradio as gr
|
9 |
+
from huggingface_hub import login
|
10 |
from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig, TextIteratorStreamer
|
11 |
|
12 |
subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)
|
13 |
|
14 |
+
login(os.environ.get("HF_TOKEN"))
|
15 |
MODEL_ID = os.environ.get("MODEL_ID")
|
16 |
CHAT_TEMPLATE = os.environ.get("CHAT_TEMPLATE")
|
17 |
MODEL_NAME = MODEL_ID.split("/")[-1]
|