Spaces:

cody82
/

bot_innopolis

Running

cody82 commited on Jul 21

Commit

de6cced

verified ·

1 Parent(s): 9d9c29a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path, local_files_only=True)
 model = AutoModelForCausalLM.from_pretrained(
     model_path,
     config=config,
-    local_files_only=True,
     torch_dtype=torch.float32,
     device_map="auto" if torch.cuda.is_available() else None,
 )

 model = AutoModelForCausalLM.from_pretrained(
     model_path,
     config=config,
+    local_files_only=False,
     torch_dtype=torch.float32,
     device_map="auto" if torch.cuda.is_available() else None,
 )