Spaces:

suwesh
/

SFAI_Agentforce_PR

Paused

suwesh commited on Feb 14

Commit

3c4359a

verified ·

1 Parent(s): 4ba67c7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ For more information on `huggingface_hub` Inference API support, please check th
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 modelpath = "Salesforce/xLAM-1b-fc-r"
-model = AutoModelForCausalLM.from_pretrained(modelpath, device_map="auto", torch_dtype="auto", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(modelpath)

 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 modelpath = "Salesforce/xLAM-1b-fc-r"
+model = AutoModelForCausalLM.from_pretrained(modelpath, torch_dtype="auto", trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(modelpath)