Spaces:

DesiredName
/

test

Build error

DesiredName commited on Jul 29

Commit

7c30434

verified ·

1 Parent(s): 4dbdc02

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,6 @@ model_name = "Tap-M/Luna-AI-Llama2-Uncensored"
 model = AutoModelForCausalLM.from_pretrained(
     model_name,  # Example model
-    device_map="auto",            # Auto-distribute across GPU/CPU
     offload_folder="./offload",  # Temporary directory
     trust_remote_code=True        # Required for some models
 )

 model = AutoModelForCausalLM.from_pretrained(
     model_name,  # Example model
     offload_folder="./offload",  # Temporary directory
     trust_remote_code=True        # Required for some models
 )