Spaces:

arasaltan
/

StudentHelper

Runtime error

arasaltan commited on Feb 13

Commit

d81f75e

verified ·

1 Parent(s): cf54705

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ LORA_PATH = "./"
 tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
 tokenizer.pad_token = tokenizer.eos_token
-# Base model (CPU)
 base_model = AutoModelForCausalLM.from_pretrained(
     repo_id,
     device_map='auto',

 tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL)
 tokenizer.pad_token = tokenizer.eos_token
+# Base model
 base_model = AutoModelForCausalLM.from_pretrained(
     repo_id,
     device_map='auto',