leonardlin commited on
Commit
ddcb276
·
verified ·
1 Parent(s): 4ad1729

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,9 +12,9 @@ DEFAULT_MAX_NEW_TOKENS = 1024
12
  MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
13
 
14
  DESCRIPTION = """\
15
- # shisa-v2-llama3.1-8b-preview
16
 
17
- This Space demonstrates the [shisa-v2-llama3.1-8b-preview](https://huggingface.co/shisa-ai/shisa-v2-llama3.1-8b-preview) bilingual (JA/EN) chat model."""
18
 
19
  LICENSE = """
20
  """
@@ -24,7 +24,7 @@ if not torch.cuda.is_available():
24
 
25
 
26
  if torch.cuda.is_available():
27
- model_id = "shisa-ai/shisa-v2-llama3.1-8b-preview"
28
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
29
  tokenizer = AutoTokenizer.from_pretrained(model_id)
30
  tokenizer.use_default_system_prompt = False
 
12
  MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
13
 
14
  DESCRIPTION = """\
15
+ # shisa-v2-llama3.1-8b
16
 
17
+ This Space demonstrates the [shisa-v2-llama3.1-8b](https://huggingface.co/shisa-ai/shisa-v2-llama3.1-8b) bilingual (JA/EN) chat model."""
18
 
19
  LICENSE = """
20
  """
 
24
 
25
 
26
  if torch.cuda.is_available():
27
+ model_id = "shisa-ai/shisa-v2-llama3.1-8b"
28
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
29
  tokenizer = AutoTokenizer.from_pretrained(model_id)
30
  tokenizer.use_default_system_prompt = False