Spaces:

AIFS
/

Prometh-MOEM-24B

Runtime error

Wtzwho commited on Feb 16, 2024

Commit

4f46af1

verified ·

1 Parent(s): 9c1f9ef

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ text_generation_pipeline = pipeline(
     "text-generation",
     model=model_name,
     model_kwargs={"torch_dtype": torch.float16, "load_in_4bit": True},
-    use_auth_token=hf_token
 )

     "text-generation",
     model=model_name,
     model_kwargs={"torch_dtype": torch.float16, "load_in_4bit": True},
+    use_auth_token=HF_TOKEN
 )