Spaces:

yasserrmd
/

Atlas-Chat

Sleeping

yasserrmd commited on Nov 7

Commit

490d5f3

•

1 Parent(s): d919dde

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,6 @@ import spaces
 model_name = "MBZUAI-Paris/Atlas-Chat-27B"
 model = AutoModelForCausalLM.from_pretrained(
-    "text-generation",
     model_name,
      model_kwargs={"torch_dtype": torch.bfloat16},
     device_map="auto"

 model_name = "MBZUAI-Paris/Atlas-Chat-27B"
 model = AutoModelForCausalLM.from_pretrained(
     model_name,
      model_kwargs={"torch_dtype": torch.bfloat16},
     device_map="auto"