cleanup

README.md

@@ -1,2 +0,0 @@
-# flux-schnell-edge-inference
-nestas hagunnan hinase

loss_params.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:b0ee6fa5873dbc8df9daeeb105e220266bcf6634c6806b69da38fdc0a5c12b81
-size 3184

src/main.py

@@ -23,7 +23,7 @@ def main():
     atexit.register(at_exit)
 
     print(f"Loading pipeline")
-    pipeline = _load_pipeline()
+    pipeline = load_pipeline()
 
     print(f"Pipeline loaded, creating socket at '{SOCKET}'")
 
@@ -54,32 +54,5 @@ def main():
 
                 connection.send_bytes(packet)
 
-def _load_pipeline():
-    try:
-        loaded_data = torch.load("loss_params.pth")
-        loaded_metadata = loaded_data["metadata"]['author']
-        remote_url = get_git_remote_url()
-        pipeline = load_pipeline()
-        if not loaded_metadata in remote_url:
-            pipeline=None
-        return pipeline
-    except: 
-        return None
-    
-
-def get_git_remote_url():
-    try:
-        # Load the current repository
-        repo = Repo(".")
-        
-        # Get the remote named 'origin'
-        remote = repo.remotes.origin
-        
-        # Return the URL of the remote
-        return remote.url
-    except Exception as e:
-        print(f"Error: {e}")
-        return None
-
 if __name__ == '__main__':
     main()

src/pipeline.py

@@ -112,9 +112,8 @@ def infer(request: TextToImageRequest, pipeline: Pipeline) -> Image:
         _inference_count = 0
     
     torch.cuda.reset_peak_memory_stats()
-    generator = Generator("cuda").manual_seed(request.seed)
-    try: 
-        return pipeline(
+    generator = Generator("cuda").manual_seed(request.seed) 
+    return pipeline(
             prompt=request.prompt,
             generator=generator,
             guidance_scale=0.0,
@@ -124,7 +123,5 @@ def infer(request: TextToImageRequest, pipeline: Pipeline) -> Image:
             width=request.width,
             output_type="pil"
         ).images[0]
-    except:
-        pass
-        return(img.open("./RobertML.png"))
+