Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
|
|
9 |
|
10 |
# Florence-2 로드
|
11 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
12 |
-
florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True).to(device).eval()
|
13 |
florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True)
|
14 |
|
15 |
# Stable Diffusion TurboX 로드
|
|
|
9 |
|
10 |
# Florence-2 로드
|
11 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
12 |
+
florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True, skip_init = True).to(device).eval()
|
13 |
florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True)
|
14 |
|
15 |
# Stable Diffusion TurboX 로드
|