Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,6 @@ REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
|
|
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, ignore_mismatched_sizes=True, ).to(device).eval()
|
13 |
-
florence_model.DaViT._initialize_weights = lambda self, module=None: None
|
14 |
florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True)
|
15 |
|
16 |
# Stable Diffusion TurboX 로드
|
|
|
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, ignore_mismatched_sizes=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 로드
|