Spaces:

Kr08
/

Llama

Sleeping

Kr08 commited on Sep 3

Commit

fc6d403

•

1 Parent(s): d690585

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ pipeline = transformers.pipeline(
     model=model_id,
     model_kwargs={"torch_dtype": torch.bfloat16},
     device_map="auto",
-    token=HF_ACCESS_TOKEN
 )
 @spaces.GPU

     model=model_id,
     model_kwargs={"torch_dtype": torch.bfloat16},
     device_map="auto",
+    token=access_token
 )
 @spaces.GPU