Spaces:

MegaTronX
/

DragonScales-Selectolax

Runtime error

MegaTronX commited on Dec 29, 2024

Commit

5f9ff8f

verified ·

1 Parent(s): 786607d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ tokenizer = AutoTokenizer.from_pretrained(base_model_name)
 model = AutoModelForCausalLM.from_pretrained(base_model_name)
 # Load your lightweight safetensor
-safetensor_path = "MegaTronX/Odyssey-SelectolaxQLoRA/adapter_model.safetensors"
 model.load_state_dict(safetensors.torch.load_file(safetensor_path), strict=False)
 @spaces.GPU

 model = AutoModelForCausalLM.from_pretrained(base_model_name)
 # Load your lightweight safetensor
+safetensor_path = "MegaTronX/Odyssey-SelectolaxQLoRA"
 model.load_state_dict(safetensors.torch.load_file(safetensor_path), strict=False)
 @spaces.GPU