Spaces:

Yumenohoshi
/

BetaAI_Chat

Runtime error

Yumenohoshi commited on Jan 20

Commit

87d74f7

•

1 Parent(s): 2c6fedd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -21,7 +21,7 @@ DESCRIPTION="""
 device = "cuda" if torch.cuda.is_available() else "cpu"
 generator = ctranslate2.Generator("./FixedStar-BETA-7b-ct2", device=device)
 tokenizer = AutoTokenizer.from_pretrained(
-    "Yumenohoshi/Fixedstar-BETA-7b", use_fast=True)
 def inference_func(prompt, max_length=128, sampling_temperature=0.7):
     tokens = tokenizer.convert_ids_to_tokens(

 device = "cuda" if torch.cuda.is_available() else "cpu"
 generator = ctranslate2.Generator("./FixedStar-BETA-7b-ct2", device=device)
 tokenizer = AutoTokenizer.from_pretrained(
+    "Yumenohoshi/FixedStar-BETA-7b", use_fast=True)
 def inference_func(prompt, max_length=128, sampling_temperature=0.7):
     tokens = tokenizer.convert_ids_to_tokens(