Spaces:

PhantHive
/

Phearion-bigbrain-v0.0.1

Paused

PhantHive commited on Feb 23, 2024

Commit

e9cf7a6

verified ·

1 Parent(s): c5ea074

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ import torch
 # Device configuration (prioritize GPU if available)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model_id = "Phearion/bigbrain-v0.0.2"
 bnb_config = BitsAndBytesConfig(
             load_in_4bit=True,

 # Device configuration (prioritize GPU if available)
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model_id = "Phearion/bigbrain-v0.0.1"
 bnb_config = BitsAndBytesConfig(
             load_in_4bit=True,