Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -6,14 +6,14 @@ import spaces
|
|
6 |
|
7 |
# Load the processor and model
|
8 |
processor = AutoProcessor.from_pretrained(
|
9 |
-
'allenai/Molmo-7B-
|
10 |
trust_remote_code=True,
|
11 |
torch_dtype='auto',
|
12 |
device_map='auto'
|
13 |
)
|
14 |
|
15 |
model = AutoModelForCausalLM.from_pretrained(
|
16 |
-
'allenai/Molmo-7B-
|
17 |
trust_remote_code=True,
|
18 |
torch_dtype='auto',
|
19 |
device_map='auto'
|
|
|
6 |
|
7 |
# Load the processor and model
|
8 |
processor = AutoProcessor.from_pretrained(
|
9 |
+
'allenai/Molmo-7B-O-0924',
|
10 |
trust_remote_code=True,
|
11 |
torch_dtype='auto',
|
12 |
device_map='auto'
|
13 |
)
|
14 |
|
15 |
model = AutoModelForCausalLM.from_pretrained(
|
16 |
+
'allenai/Molmo-7B-O-0924',
|
17 |
trust_remote_code=True,
|
18 |
torch_dtype='auto',
|
19 |
device_map='auto'
|