Spaces:

aifeifei798
/

SmolVLM-Image-To-Flux-Prompt

Running on Zero

aifeifei798 commited on about 13 hours ago

Commit

2415d69

verified ·

1 Parent(s): de92b09

Update feifeilib/feifeismolvlm.py

Files changed (1) hide show

feifeilib/feifeismolvlm.py CHANGED Viewed

@@ -7,8 +7,8 @@ import time
 import torch
 import spaces
-processor = AutoProcessor.from_pretrained("HuggingFaceTB/SmolVLM2-256M-Video-Instruct")
-model = AutoModelForVision2Seq.from_pretrained("HuggingFaceTB/SmolVLM2-256M-Video-Instruct",
         torch_dtype=torch.bfloat16,
         #_attn_implementation="flash_attention_2"
         ).to("cuda")

 import torch
 import spaces
+processor = AutoProcessor.from_pretrained("HuggingFaceTB/SmolVLM-256M-Instruct")
+model = AutoModelForVision2Seq.from_pretrained("HuggingFaceTB/SmolVLM-256M-Instruct",
         torch_dtype=torch.bfloat16,
         #_attn_implementation="flash_attention_2"
         ).to("cuda")